arXiv Computer Vision By Ba-Thinh Lam, Srijan Das, Hieu Le

Importance-Aware OBS Pruning for Diffusion Models

Read the original on arXiv Computer Vision →

The paper introduces an importance-aware pruning method for diffusion models that is training‑free and focuses on preserving parameters essential to semantically salient image regions. By integrating spatial importance maps derived from conditioning signals or model attention into the pruning objective, the method generates parameter rankings that reflect perceptual relevance rather than uniform reconstruction error. Experiments on the MS‑COCO dataset show that this approach maintains subject fidelity and structural correctness even at high compression ratios, outperforming conventional pruning techniques that cause visible degradation.

Machine-generated by The Flow from the publisher's headline and feed description — not written or checked by a human. The full article lives at arXiv Computer Vision.

arXiv Machine Learning
Sep 25

CARE: Condition-Aware Representation Regularization for Diffusion Models

The paper introduces CARE, a lightweight, plug‑and‑play regularization framework for diffusion models that dynamically adjusts feature distributions based on condition similarity. By leveraging built‑in conditioning signals such as labels or text prompts, CARE promotes tighter feature clusters for similar conditions without requiring explicit alignment losses or external supervision. Empirical results show consistent improvements in visual fidelity and convergence stability, achieving significant FID reductions and speed‑ups on ImageNet and text‑to‑image tasks, and it can be combined with existing regularization methods for further gains.

By Fengjia Guo, Zhuoyi Yang, Jie Tang