Targeted Layer Retention for Mixed-Precision NF4 Diffusion Transformers

This preprint documents a targeted layer-retention method for mixed-precision NF4 quantization of diffusion-transformer image models. The method retains architecture-specific first, last, and boundary modules in bfloat16 while quantizing the middle transformer body to 4-bit NormalFloat (NF4) with double quantization. The central claim is that useful compression depends not only on using NF4, but on selecting which layers should not be quantized. Public Hugging Face artifacts demonstrate this rule across text-conditioned image editing with Qwen-Image-Edit variants and text-to-image generation with ERNIE-Image. The first related public artifact, ovedrive/qwen-image-edit-4bit, was released on Hugging Face on 19 August 2025. This Zenodo record archives the manuscript describing the method, artifact history, deployment observations, limitations, and future benchmark directions.

Authors

Publication Details

Journal
Zenodo (CERN European Organization for Nuclear Research)
Published
2026-05-27
DOI
https://doi.org/10.5281/zenodo.20405685
Primary Topic
Cell Image Analysis Techniques
Type
preprint
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
preprint

Targeted Layer Retention for Mixed-Precision NF4 Diffusion Transformers

Abhishek Dujari
Zenodo (CERN European Organization for Nuclear Research)
Cell Image Analysis Techniques
preprint

Targeted Layer Retention for Mixed-Precision NF4 Diffusion Transformers

Abhishek Dujari
preprint en

Abstract

This preprint documents a targeted layer-retention method for mixed-precision NF4 quantization of diffusion-transformer image models. The method retains architecture-specific first, last, and boundary modules in bfloat16 while quantizing the middle transformer body to 4-bit NormalFloat (NF4) with double quantization. The central claim is that useful compression depends not only on using NF4, but on selecting which layers should not be quantized. Public Hugging Face artifacts demonstrate this rule across text-conditioned image editing with Qwen-Image-Edit variants and text-to-image generation with ERNIE-Image. The first related public artifact, ovedrive/qwen-image-edit-4bit, was released on Hugging Face on 19 August 2025. This Zenodo record archives the manuscript describing the method, artifact history, deployment observations, limitations, and future benchmark directions.

Zenodo (CERN European Organization for Nuclear Research)
Cell Image Analysis Techniques
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.