paper-with-me

홈 › Papers

Routing the Lottery: Adaptive Subnetworks for Heterogeneous Data

2026-01-29 · Grzegorz Stefanski, Alberto Presta, Michal Byra arxiv

In pruning, the Lottery Ticket Hypothesis posits that large networks contain sparse subnetworks, or winning tickets, that can be trained in isolation to match the performance of their dense counterparts. However, most existing approaches assume a single universal winning ticket shared across all inputs, ignoring the inherent heterogeneity of real-world data. In this work, we propose Routing the Lottery (RTL), an adaptive pruning framework that discovers multiple specialized subnetworks, called adaptive tickets, each tailored to a class, semantic cluster, or environmental condition. Across diverse datasets and tasks, RTL consistently outperforms single- and multi-model baselines in balanced accuracy and recall, while using up to 10 times fewer parameters than independent models and exhibiting semantically aligned. Furthermore, we identify subnetwork collapse, a performance drop under aggressive pruning, and introduce a subnetwork similarity score that enables label-free diagnosis of oversparsification. Overall, our results recast pruning as a mechanism for aligning model structure with data heterogeneity, paving the way toward more modular and context-aware deep learning.

📄 PDF Abstract BibTeX arXiv:2601.22141

Code (0)

등록된 구현이 없습니다.

Similar Papers 제목 키워드 기반

Forget-free Continual Learning with Winning Subnetworks

2022-07-01 · International Conference on Machine Learning 2022 7 · Haeyong Kang, Rusty John Lloyd Mina, Sultan Rizky Hikmawan Madjid, Jaehong Yoon 외

Inspired by Lottery Ticket Hypothesis that competitive subnetworks exist within a dense network, we propose a continual learning method referred to as Winning SubNetworks (WSN), which sequentially learns and selects an o…

Continual Learning

Evaluating Lottery Tickets Under Distributional Shifts

2019-10-28 · WS 2019 11 · Shrey Desai, Hongyuan Zhan, Ahmed Aly

The Lottery Ticket Hypothesis suggests large, over-parameterized neural networks consist of small, sparse subnetworks that can be trained in isolation to reach a similar (or better) test accuracy. However, the initializa…

Inductive Bias

Lottery Pools: Winning More by Interpolating Tickets without Increasing Training or Inference Cost

2022-08-23 · Lu Yin, Shiwei Liu, Meng Fang, Tianjin Huang 외

Lottery tickets (LTs) is able to discover accurate and sparse subnetworks that could be trained in isolation to match the performance of dense networks. Ensemble, in parallel, is one of the oldest time-proven tricks in m…

Data-Efficient Double-Win Lottery Tickets from Robust Pre-training

2022-06-09 · Tianlong Chen, Zhenyu Zhang, Sijia Liu, Yang Zhang 외

Pre-training serves as a broadly adopted starting point for transfer learning on various downstream tasks. Recent investigations of lottery tickets hypothesis (LTH) demonstrate such enormous pre-trained models can be rep…

Transfer Learning

The Lottery Ticket Hypothesis for Pre-trained BERT Networks

2020-07-23 · NeurIPS 2020 12 · Tianlong Chen, Jonathan Frankle, Shiyu Chang, Sijia Liu 외

In natural language processing (NLP), enormous pre-trained models like BERT have become the standard starting point for training on a range of downstream tasks, and similar trends are emerging in other areas of deep lear…

Language ModelingLanguage ModellingMasked Language Modeling