mlsanityData to 7 Oct 2026

Muon-Accelerated Attention Distillation for Real-Time Edge Synthesis via Optimized Latent Diffusion

arXiv 2504.08451 ↗First posted 11 Apr 2025cs.CVcited by 2

On the map: Diffusion models → Diffusion foundations and sampling

Abstract (arXiv)

Recent advances in visual synthesis have leveraged diffusion models and attention mechanisms to achieve high-fidelity artistic style transfer and photorealistic text-to-image generation. However, real-time deployment on edge devices remains challenging due to computational and memory constraints. We propose Muon-AD, a co-designed framework that integrates the Muon optimizer with attention distillation for real-time edge synthesis. By eliminating gradient conflicts through orthogonal parameter updates and dynamic pruning, Muon-AD achieves 3.2 times faster convergence compared to Stable Diffusion-TensorRT, while maintaining synthesis quality (15% lower FID, 4% higher SSIM). Our framework reduces peak memory to 7GB on Jetson Orin and enables 24FPS real-time generation through mixed-precision quantization and curriculum learning. Extensive experiments on COCO-Stuff and ImageNet-Texture demonstrate Muon-AD's Pareto-optimal efficiency-quality trade-offs. Here, we show a 65% reduction in communication overhead during distributed training and real-time 10s/image generation on edge GPUs. These advancements pave the way for democratizing high-quality visual synthesis in resource-constrained environments.

Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Placed in Diffusion foundations and sampling, one of 14 directions in Diffusion models. Builds on 7 of its 37 references that are in the map. First posted on arXiv 11 Apr 2025.

Builds on

7 of 37 references.

Closest ancestors · 1

Same direction, newest first

  1. 2021High-Resolution Image Synthesis with Latent Diffusion Models20 Dec

Other references · 6

  1. 2014Adam: A Method for Stochastic Optimizationcited by 36,960
  2. 2014Microsoft COCO: Common Objects in Contextcited by 17,498
  3. 2019ZeRO: Memory Optimizations Toward Training Trillion Parameter Modelscited by 1,278
  4. 2017Mixed Precision Trainingcited by 1,135
  5. 2025Attention Distillation: A Unified Approach to Visual Characteristics Transfercited by 32
  6. 2023DynaMask: Dynamic Mask Selection for Instance Segmentationcited by 10

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 2.

2 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2025Muon: Training and Trade-offs with Latent Attention and MoE
  2. 2026CMuon: Accelerating and Stabilizing Diffusion Transformer Training via Chunked Momentum Orthogonalization
  3. 2026Scaling Muon for Diffusion Transformers
  4. 2025Grouping First, Attending Smartly: Training-Free Acceleration for Diffusion Transformers
  5. 2026SnapGen++: Unleashing Diffusion Transformers for Efficient High-Fidelity Image Generation on Edge Devices
  6. 2026Rethinking Muon Beyond Pretraining: Spectral Failures and High-Pass Remedies for VLA and RLVR
  7. 2025ToMA: Token Merge with Attention for Diffusion Models
  8. 2025UltraImageGen: Efficient Ultra-High-Resolution Image Generation with Hierarchical Local Attention