PruneFuse: Efficient Data Selection via Weight Pruning and Network Fusion

View PDF HTML (experimental)

Abstract:Efficient data selection is crucial for enhancing the training efficiency of deep neural networks and minimizing annotation requirements. Traditional methods often face high computational costs, limiting their scalability and practical use. We introduce PruneFuse, a novel strategy that leverages pruned networks for data selection and later fuses them with the original network to optimize training. PruneFuse operates in two stages: First, it applies structured pruning to create a smaller pruned network that, due to its structural coherence with the original network, is well-suited for the data selection task. This small network is then trained and selects the most informative samples from the dataset. Second, the trained pruned network is seamlessly fused with the original network. This integration leverages the insights gained during the training of the pruned network to facilitate the learning process of the fused network while leaving room for the network to discover more robust solutions. Extensive experimentation on various datasets demonstrates that PruneFuse significantly reduces computational costs for data selection, achieves better performance than baselines, and accelerates the overall training process.

Comments: Published in TMLR (Featured Certification). arXiv admin note: substantial text overlap with arXiv:2501.01118

Subjects:

Machine Learning (cs.LG); Computer Vision and Pattern Recognition (cs.CV)

Cite as: arXiv:2603.26138 [cs.LG]

(or arXiv:2603.26138v1 [cs.LG] for this version)

https://doi.org/10.48550/arXiv.2603.26138

arXiv-issued DOI via DataCite (pending registration)

Journal reference: Transactions on Machine Learning Research (TMLR), March 2026

Submission history

From: Hasnain Irshad Bhatti [view email] [v1] Fri, 27 Mar 2026 07:46:28 UTC (2,794 KB)