Annotation-Efficient Coreset Selection for Context-dependent Segmentation
Jin Zhang, Zhe Cao, Biwen Yang, Ruiheng Zhang
Abstract
Context-dependent (CD) tasks demand the model to have advanced visual understanding ability, such as recognizing camouflaged objects and medical lesions. Current CD methods rely heavily on pixel-level annotated training sets, neglecting issues from redundant samples and the high annotation costs. In this paper, we address the pruning needs of CD datasets, focusing on selecting the most valuable samples for labeling and training using weak annotations. To achieve this, we decompose CD coreset selection into two steps: sample evaluation and coreset selection, proposing corresponding solutions: points-based optimal transport and a maximum distance entropy strategy. Specifically, we formulate sample evaluation as an optimal transport problem between foreground and background distributions, designing a foreground destruction-reconstruction process based on points to compute transport costs and score samples. For samples of varying importance, our selection strategy balances coreset coverage and diversity. We validate our method on six CD tasks, achieving 1% accuracy loss relative to full training under a 40% pruning rate.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 40374f24-9bd4-4654-839a-606ef1e31d20Builds on23
- Denoising Diffusion Probabilistic ModelsJonathan Ho, Ajay Jain, Pieter AbbeelNeurIPS 2020 · 35,902 citations
- Deep Learning on a Data Diet: Finding Important Examples Early in TrainingMansheej Paul, Surya Ganguli, Gintare Karolina DziugaiteNeurIPS 2021 · 806 citations
- Selection via Proxy: Efficient Data Selection for Deep LearningCody Coleman, Christopher Yeh, Stephen Mussmann, Baharan Mirzasoleiman et al.ICLR 2020 · 462 citations
- GRAD-MATCH: Gradient Matching based Data Subset Selection for Efficient Deep Model TrainingKrishnaTeja Killamsetty, Durga Sivasubramanian, Ganesh Ramakrishnan, Abir De et al.ICML 2021 · 305 citations
- Faster Wasserstein Distance Estimation with the Sinkhorn DivergenceLénaïc Chizat, Pierre Roussillon, Flavien Léger, François-Xavier Vialard et al.NeurIPS 2020 · 164 citations
Related papers
- Point2Mask: Point-supervised Panoptic Segmentation via Optimal TransportWentong Li, Yuqian Yuan, Song Wang, Jianke Zhu et al.ICCV 2023 · 34 citations
- Coverage-centric Coreset Selection for High Pruning RatesHaizhong Zheng, Rui Liu, Fan Lai, Atul PrakashICLR 2023 · 3 citations
- Feature Distribution Matching by Optimal Transport for Effective and Robust Coreset SelectionWeiwei Xiao, Yongyong Chen, Qiben Shan, Yaowei Wang et al.AAAI 2024 · 14 citations
- Evolution-aware VAriance (EVA) Coreset Selection for Medical Image ClassificationYuxin Hong, Xiao Zhang, Xin Zhang, Joey Tianyi ZhouACM MM 2024 · 4 citations
- FOCT: Few-shot Industrial Anomaly Detection with Foreground-aware Online Conditional TransportLong Tian, Hongyi Zhao, Ruiying Lu, Rongrong Wang et al.ACM MM 2024 · 6 citations
