RS-Prune: Training-Free Data Pruning at High Ratios for Efficient Remote Sensing Diffusion Foundation Models

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Wei, Fan, Dong, Runmin, Lai, Yushan, Yang, Yixiang, Luo, Zhaoyang, Zhang, Jinxiao, Yang, Miao, Yuan, Shuai, Zhao, Jiyao, Luo, Bin, Fu, Haohuan
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915706456506368
author Wei, Fan
Dong, Runmin
Lai, Yushan
Yang, Yixiang
Luo, Zhaoyang
Zhang, Jinxiao
Yang, Miao
Yuan, Shuai
Zhao, Jiyao
Luo, Bin
Fu, Haohuan
author_facet Wei, Fan
Dong, Runmin
Lai, Yushan
Yang, Yixiang
Luo, Zhaoyang
Zhang, Jinxiao
Yang, Miao
Yuan, Shuai
Zhao, Jiyao
Luo, Bin
Fu, Haohuan
contents Diffusion-based remote sensing (RS) generative foundation models are cruial for downstream tasks. However, these models rely on large amounts of globally representative data, which often contain redundancy, noise, and class imbalance, reducing training efficiency and preventing convergence. Existing RS diffusion foundation models typically aggregate multiple classification datasets or apply simplistic deduplication, overlooking the distributional requirements of generation modeling and the heterogeneity of RS imagery. To address these limitations, we propose a training-free, two-stage data pruning approach that quickly select a high-quality subset under high pruning ratios, enabling a preliminary foundation model to converge rapidly and serve as a versatile backbone for generation, downstream fine-tuning, and other applications. Our method jointly considers local information content with global scene-level diversity and representativeness. First, an entropy-based criterion efficiently removes low-information samples. Next, leveraging RS scene classification datasets as reference benchmarks, we perform scene-aware clustering with stratified sampling to improve clustering effectiveness while reducing computational costs on large-scale unlabeled data. Finally, by balancing cluster-level uniformity and sample representativeness, the method enables fine-grained selection under high pruning ratios while preserving overall diversity and representativeness. Experiments show that, even after pruning 85\% of the training data, our method significantly improves convergence and generation quality. Furthermore, diffusion foundation models trained with our method consistently achieve state-of-the-art performance across downstream tasks, including super-resolution and semantic image synthesis. This data pruning paradigm offers practical guidance for developing RS generative foundation models.
format Preprint
id arxiv_https___arxiv_org_abs_2512_23239
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle RS-Prune: Training-Free Data Pruning at High Ratios for Efficient Remote Sensing Diffusion Foundation Models
Wei, Fan
Dong, Runmin
Lai, Yushan
Yang, Yixiang
Luo, Zhaoyang
Zhang, Jinxiao
Yang, Miao
Yuan, Shuai
Zhao, Jiyao
Luo, Bin
Fu, Haohuan
Computer Vision and Pattern Recognition
Diffusion-based remote sensing (RS) generative foundation models are cruial for downstream tasks. However, these models rely on large amounts of globally representative data, which often contain redundancy, noise, and class imbalance, reducing training efficiency and preventing convergence. Existing RS diffusion foundation models typically aggregate multiple classification datasets or apply simplistic deduplication, overlooking the distributional requirements of generation modeling and the heterogeneity of RS imagery. To address these limitations, we propose a training-free, two-stage data pruning approach that quickly select a high-quality subset under high pruning ratios, enabling a preliminary foundation model to converge rapidly and serve as a versatile backbone for generation, downstream fine-tuning, and other applications. Our method jointly considers local information content with global scene-level diversity and representativeness. First, an entropy-based criterion efficiently removes low-information samples. Next, leveraging RS scene classification datasets as reference benchmarks, we perform scene-aware clustering with stratified sampling to improve clustering effectiveness while reducing computational costs on large-scale unlabeled data. Finally, by balancing cluster-level uniformity and sample representativeness, the method enables fine-grained selection under high pruning ratios while preserving overall diversity and representativeness. Experiments show that, even after pruning 85\% of the training data, our method significantly improves convergence and generation quality. Furthermore, diffusion foundation models trained with our method consistently achieve state-of-the-art performance across downstream tasks, including super-resolution and semantic image synthesis. This data pruning paradigm offers practical guidance for developing RS generative foundation models.
title RS-Prune: Training-Free Data Pruning at High Ratios for Efficient Remote Sensing Diffusion Foundation Models
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2512.23239