Method · Gradient matching
Low-Level Dataset Distillation for Medical Image Enhancement
Fengzhi Xu, Ziyuan Yang, Mengyu Sun, Joey Tianyi Zhou, Yi Zhang
arXiv 2025 · first public 2025-11-17 · arXiv 2511.13106
In one paragraph
Argues that low-level tasks (denoising, enhancement) make dataset distillation underdetermined because the many-to-many pixel-level mapping cannot be constrained by a small distilled set the way many-to-one classification labels can; proposes the first low-level DD method, which builds a shared anatomical prior from one representative patient, personalizes it per patient with a Structure-Preserving Personalized Generation module, and injects patient-specific knowledge by matching gradients between networks trained on distilled pairs and on the patient's raw data. Downstream users only receive the abstracted distilled pairs, not raw patient data.
Where it sits
- Gradient matching (Surrogate matching)
- Setting: Detection, segmentation and low-level vision
Abstract (verbatim from arXiv)
Medical image enhancement is clinically valuable, but existing methods require large-scale datasets to learn complex pixel-level mappings. However, the substantial training and storage costs associated with these datasets hinder their practical deployment. While dataset distillation (DD) can alleviate these burdens, existing methods mainly target high-level tasks, where multiple samples share the same label. This many-to-one mapping allows distilled data to capture shared semantics and achieve information compression. In contrast, low-level tasks involve a many-to-many mapping that requires pixel-level fidelity, making low-level DD an underdetermined problem, as a small distilled dataset cannot fully constrain the dense pixel-level mappings. To address this, we propose the first low-level DD method for medical image enhancement. We first leverage anatomical similarities across patients to construct the shared anatomical prior based on a representative patient, which serves as the initialization for the distilled data of different patients. This prior is then personalized for each patient using a Structure-Preserving Personalized Generation (SPG) module, which integrates patient-specific anatomical information into the distilled dataset while preserving pixel-level fidelity. For different low-level tasks, the distilled data is used to construct task-specific high- and low-quality training pairs. Patient-specific knowledge is injected into the distilled data by aligning the gradients computed from networks trained on the distilled pairs with those from the corresponding patient's raw data. Notably, downstream users cannot access raw patient data. Instead, only a distilled dataset containing abstract training information is shared, which excludes patient-specific details and thus preserves privacy.
BibTeX (generated; prefer the venue's official entry)
@article{xu2025level,
title = {Low-Level Dataset Distillation for Medical Image Enhancement},
author = {Fengzhi Xu and Ziyuan Yang and Mengyu Sun and Joey Tianyi Zhou and Yi Zhang},
journal = {arXiv preprint arXiv:2511.13106},
year = {2025}
}Nearby in Gradient matching
Linear Gradient Matching — Dataset Distillation for Pre-Trained Self-Supervised Vision Models
George Cazenavette, Antonio Torralba, Vincent Sitzmann · NeurIPS 2025notablePre-training & transferpaper ↗code ↗
Distilled Datamodel with Reverse Gradient Matching
Jingwen Ye, Ruonan Yu, Songhua Liu et al. · CVPR 2024notablepaper ↗
Static-dynamic video DD — Dancing with Still Images: Video Distillation via Static-Dynamic Disentanglement
Ziyu Wang, Yue Xu, Cewu Lu et al. · CVPR 2024coreVideopaper ↗code ↗