Condition-Aware Neural Network for Controlled Image Generation
Not on the map. Why?
Abstract (arXiv)
We present Condition-Aware Neural Network (CAN), a new method for adding control to image generative models. In parallel to prior conditional control methods, CAN controls the image generation process by dynamically manipulating the weight of the neural network. This is achieved by introducing a condition-aware weight generation module that generates conditional weight for convolution/linear layers based on the input condition. We test CAN on class-conditional image generation on ImageNet and text-to-image generation on COCO. CAN consistently delivers significant improvements for diffusion transformer models, including DiT and UViT. In particular, CAN combined with EfficientViT (CaT) achieves 2.78 FID on ImageNet 512x512, surpassing DiT-XL/2 while requiring 52x fewer MACs per sampling step.
Key facts. Cited by 16 papers in the map (data through 14 Oct 2026). Builds on 32 of its 37 references that are in the map. First posted on arXiv 1 Apr 2024.
Builds on
32 of 37 references.
References · 32
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
- 2020Denoising Diffusion Probabilistic Modelscited by 21,076
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,863
- 2021Diffusion Models Beat GANs on Image Synthesiscited by 8,255
- 2022Photorealistic Text-to-Image Diffusion Models with Deep Language Understandingcited by 6,355
- 2022Scalable Diffusion Models with Transformerscited by 6,054
- 2018A Style-Based Generator Architecture for Generative Adversarial Networkscited by 6,013
- 2022Classifier-Free Diffusion Guidancecited by 5,847
- 2023Adding Conditional Control to Text-to-Image Diffusion Modelscited by 5,503
- 2019Analyzing and Improving the Image Quality of StyleGANcited by 3,588
- 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,509
- 2018Large Scale GAN Training for High Fidelity Natural Image Synthesiscited by 2,626
20 more
- 2023Stable Video Diffusion: Scaling Latent Video Diffusion Models to Large Datasetscited by 2,581
- 2021CLIPScore: A Reference-free Evaluation Metric for Image Captioningcited by 2,424
- 2017FiLM: Visual Reasoning with a General Conditioning Layercited by 2,333
- 2016Xception: Deep Learning with Depthwise Separable Convolutionscited by 2,332
- 2015Deep Compression: Compressing Deep Neural Networks with Pruning, Trained Quantization and Huffman Codingcited by 2,329
- 2017Arbitrary Style Transfer in Real-time with Adaptive Instance Normalizationcited by 2,010
- 2022DPM-Solver: A Fast ODE Solver for Diffusion Probabilistic Model Sampling in Around 10 Stepscited by 1,979
- 2023Consistency Modelscited by 1,822
- 2016HyperNetworkscited by 1,027
- 2019Once-for-All: Train One Network and Specialize it for Efficient Deploymentcited by 787
- 2018ProxylessNAS: Direct Neural Architecture Search on Target Task and Hardwarecited by 673
- 2023Scaling up GANs for Text-to-Image Synthesiscited by 492
- 2023UniPC: A Unified Predictor-Corrector Framework for Fast Sampling of Diffusion Modelscited by 450
- 2018Slimmable Neural Networkscited by 305
- 2019CondConv: Conditionally Parameterized Convolutions for Efficient Inferencecited by 273
- 2023Q-Diffusion: Quantizing Diffusion Modelscited by 273
- 2017SMASH: One-Shot Model Architecture Search through HyperNetworkscited by 202
- 2020Understanding and Improving Information Transfer in Multi-Task Learningcited by 114
- 2020Improving Image Captioning with Better Use of Captionscited by 84
- 2021Network Augmentation for Tiny Deep Learningcited by 24
Built on by
Citing papers per year: 2024: 5; 2025: 8; 2026: 3.
16 papers.
Sort:
By area:
- 2024SANA: Efficient High-Resolution Image Synthesis with Linear Diffusion Transformerscited by 340
- 2024Deep Compression Autoencoder for Efficient High-Resolution Diffusion Modelscited by 217
- 2025Sparse VideoGen: Accelerating Video Diffusion Transformers with Spatial-Temporal Sparsitycited by 217
- 2024SVDQuant: Absorbing Outliers by Low-Rank Components for 4-Bit Diffusion Modelscited by 170
- 2025SANA-Sprint: One-Step Diffusion with Continuous-Time Consistency Distillationcited by 87
- 2024FasterDiT: Towards Faster Diffusion Transformers Training without Architecture Modificationcited by 64
- 2025DC-AE 1.5: Accelerating Diffusion Model Convergence with Structured Latent Spacecited by 47
- 2025Jet-Nemotron: Efficient Language Model with Post Neural Architecture Searchcited by 41
- 2025DC-AR: Efficient Masked Autoregressive Image Generation with Deep Compression Hybrid Tokenizercited by 13
- 2025LiT: Delving into a Simple Linear Diffusion Transformer for Image Generationcited by 10
- 2025DC-Gen: Post-Training Diffusion Acceleration with Deeply Compressed Latent Spacecited by 8
- 2024CASC: Condition-Aware Semantic Communication with Latent Diffusion Modelscited by 0
4 more
- 2025In the Blink of an Eye: Instant Game Map Editing using a Generative-AI Smart Brushcited by 0
- 2026JetViT: Efficient High-Resolution Vision Transformer with Post-Training Attention Searchcited by 0
- 2026SmartFont: Dynamic Condition Allocation for Few-Shot Font Generationcited by 0
- 2026Space Generative AI with Solar Energy Harvestingcited by 0
- 2026Space Generative AI with Solar Energy Harvesting1 Sep 2026
- 2026SmartFont: Dynamic Condition Allocation for Few-Shot Font Generation11 Jun 2026
- 2026JetViT: Efficient High-Resolution Vision Transformer with Post-Training Attention Search26 May 2026
- 2025DC-Gen: Post-Training Diffusion Acceleration with Deeply Compressed Latent Space29 Sep 2025
- 2025Jet-Nemotron: Efficient Language Model with Post Neural Architecture Search21 Aug 2025
- 2025DC-AE 1.5: Accelerating Diffusion Model Convergence with Structured Latent Space1 Aug 2025
- 2025DC-AR: Efficient Masked Autoregressive Image Generation with Deep Compression Hybrid Tokenizer7 Jul 2025
- 2025In the Blink of an Eye: Instant Game Map Editing using a Generative-AI Smart Brush25 Mar 2025
- 2025SANA-Sprint: One-Step Diffusion with Continuous-Time Consistency Distillation12 Mar 2025
- 2025Sparse VideoGen: Accelerating Video Diffusion Transformers with Spatial-Temporal Sparsity3 Feb 2025
- 2025LiT: Delving into a Simple Linear Diffusion Transformer for Image Generation22 Jan 2025
- 2024CASC: Condition-Aware Semantic Communication with Latent Diffusion Models10 Nov 2024
4 more
- 2024SVDQuant: Absorbing Outliers by Low-Rank Components for 4-Bit Diffusion Models7 Nov 2024
- 2024Deep Compression Autoencoder for Efficient High-Resolution Diffusion Models14 Oct 2024
- 2024SANA: Efficient High-Resolution Image Synthesis with Linear Diffusion Transformers14 Oct 2024
- 2024FasterDiT: Towards Faster Diffusion Transformers Training without Architecture Modification14 Oct 2024
- 2024SANA: Efficient High-Resolution Image Synthesis with Linear Diffusion Transformerscited by 340
- 2024Deep Compression Autoencoder for Efficient High-Resolution Diffusion Modelscited by 217
- 2025Sparse VideoGen: Accelerating Video Diffusion Transformers with Spatial-Temporal Sparsitycited by 217
- 2024SVDQuant: Absorbing Outliers by Low-Rank Components for 4-Bit Diffusion Modelscited by 170
- 2025SANA-Sprint: One-Step Diffusion with Continuous-Time Consistency Distillationcited by 87
- 2024FasterDiT: Towards Faster Diffusion Transformers Training without Architecture Modificationcited by 64
- 2025DC-AE 1.5: Accelerating Diffusion Model Convergence with Structured Latent Spacecited by 47
- 2025LiT: Delving into a Simple Linear Diffusion Transformer for Image Generationcited by 10
- 2025DC-Gen: Post-Training Diffusion Acceleration with Deeply Compressed Latent Spacecited by 8
- 2024CASC: Condition-Aware Semantic Communication with Latent Diffusion Modelscited by 0
- 2025DC-Gen: Post-Training Diffusion Acceleration with Deeply Compressed Latent Space29 Sep 2025
- 2025DC-AE 1.5: Accelerating Diffusion Model Convergence with Structured Latent Space1 Aug 2025
- 2025SANA-Sprint: One-Step Diffusion with Continuous-Time Consistency Distillation12 Mar 2025
- 2025Sparse VideoGen: Accelerating Video Diffusion Transformers with Spatial-Temporal Sparsity3 Feb 2025
- 2025LiT: Delving into a Simple Linear Diffusion Transformer for Image Generation22 Jan 2025
- 2024CASC: Condition-Aware Semantic Communication with Latent Diffusion Models10 Nov 2024
- 2024SVDQuant: Absorbing Outliers by Low-Rank Components for 4-Bit Diffusion Models7 Nov 2024
- 2024Deep Compression Autoencoder for Efficient High-Resolution Diffusion Models14 Oct 2024
- 2024SANA: Efficient High-Resolution Image Synthesis with Linear Diffusion Transformers14 Oct 2024
- 2024FasterDiT: Towards Faster Diffusion Transformers Training without Architecture Modification14 Oct 2024
- 2025Jet-Nemotron: Efficient Language Model with Post Neural Architecture Searchcited by 41
- 2025DC-AR: Efficient Masked Autoregressive Image Generation with Deep Compression Hybrid Tokenizercited by 13
- 2025In the Blink of an Eye: Instant Game Map Editing using a Generative-AI Smart Brushcited by 0
- 2026JetViT: Efficient High-Resolution Vision Transformer with Post-Training Attention Searchcited by 0
- 2026SmartFont: Dynamic Condition Allocation for Few-Shot Font Generationcited by 0
- 2026Space Generative AI with Solar Energy Harvestingcited by 0
- 2026Space Generative AI with Solar Energy Harvesting1 Sep 2026
- 2026SmartFont: Dynamic Condition Allocation for Few-Shot Font Generation11 Jun 2026
- 2026JetViT: Efficient High-Resolution Vision Transformer with Post-Training Attention Search26 May 2026
- 2025Jet-Nemotron: Efficient Language Model with Post Neural Architecture Search21 Aug 2025
- 2025DC-AR: Efficient Masked Autoregressive Image Generation with Deep Compression Hybrid Tokenizer7 Jul 2025
- 2025In the Blink of an Eye: Instant Game Map Editing using a Generative-AI Smart Brush25 Mar 2025
Similar, not cited
Close in wording, no citation either way
- 2024A Simple Approach to Unifying Diffusion-based Conditional Generation
- 2024CtrLoRA: An Extensible and Efficient Framework for Controllable Image Generation
- 2025CPO: Condition Preference Optimization for Controllable Image Generation
- 2023CoDi: Conditional Diffusion Distillation for Higher-Fidelity and Faster Image Generation
- 2023Conditional Image Generation with Pretrained Generative Model
- 2022On Conditioning the Input Noise for Controlled Image Generation with Diffusion Models
- 2024DynamicControl: Adaptive Condition Selection for Improved Text-to-Image Generation
- 2023Conditional Generation from Unconditional Diffusion Models using Denoiser Representations