AI Navigate

連続緩和ベルヌーイゲートによる勝利の宝くじチケットの発見

arXiv cs.AI / 2026/3/11

Ideas & Deep AnalysisModels & Research

要点

  • 本論文は、重みの学習なしにStrong Lottery Tickets(SLTs)と呼ばれるスパースなサブネットワークを発見することで、リソース制約のあるデバイスでの過パラメータ化ニューラルネットワークの展開という課題に取り組んでいる。
  • 著者らは、連続緩和ベルヌーイゲートを用いた、完全微分可能なエンドツーエンド最適化という新しい手法を提案し、ネットワークの重みを固定したままゲーティングパラメータのみを訓練する。
  • このアプローチは、edge-popupのような既存手法で用いられる非微分可能な勾配推定器や反復的なプルーニングサイクルを回避し、$9$l_0$正則化目的関数の直接的な勾配ベース最適化を可能にする。
  • 実験では、全結合ネットワーク、ResNets、Wide-ResNets、Vision Transformers、Swin-Tモデルを含む多様なアーキテクチャで最大90%のネットワークスパース性を達成し、精度低下は最小限であり、従来の研究と比べてほぼ2倍のスパース化を実現している。
  • 提案されたフレームワークは、リソース制約環境での効率的展開に不可欠なニューラルネットワークの事前スパース化のためのスケーラブルかつ効率的な方法を確立している。

Computer Science > Machine Learning

arXiv:2603.08914 (cs)
[Submitted on 9 Mar 2026]

Title:Uncovering a Winning Lottery Ticket with Continuously Relaxed Bernoulli Gates

View a PDF of the paper titled Uncovering a Winning Lottery Ticket with Continuously Relaxed Bernoulli Gates, by Itamar Tsayag and 1 other authors
View PDF HTML (experimental)
Abstract:Over-parameterized neural networks incur prohibitive memory and computational costs for resource-constrained deployment. The Strong Lottery Ticket (SLT) hypothesis suggests that randomly initialized networks contain sparse subnetworks achieving competitive accuracy without weight training. Existing SLT methods, notably edge-popup, rely on non-differentiable score-based selection, limiting optimization efficiency and scalability. We propose using continuously relaxed Bernoulli gates to discover SLTs through fully differentiable, end-to-end optimization - training only gating parameters while keeping all network weights frozen at their initialized values. Continuous relaxation enables direct gradient-based optimization of an $\ell_0$-regularization objective, eliminating the need for non-differentiable gradient estimators or iterative pruning cycles. To our knowledge, this is the first fully differentiable approach for SLT discovery that avoids straight-through estimator approximations. Experiments across fully connected networks, CNNs (ResNet, Wide-ResNet), and Vision Transformers (ViT, Swin-T) demonstrate up to 90% sparsity with minimal accuracy loss - nearly double the sparsity achieved by edge-popup at comparable accuracy - establishing a scalable framework for pre-training network sparsification.
Subjects: Machine Learning (cs.LG); Artificial Intelligence (cs.AI)
Cite as: arXiv:2603.08914 [cs.LG]
  (or arXiv:2603.08914v1 [cs.LG] for this version)
  https://doi.org/10.48550/arXiv.2603.08914
Focus to learn more
arXiv-issued DOI via DataCite

Submission history

From: Itamar Tsayag Mr [view email]
[v1] Mon, 9 Mar 2026 20:33:16 UTC (570 KB)
Full-text links:

Access Paper:

Current browse context:
cs.LG
< prev   |   next >
Change to browse by:

References & Citations

export BibTeX citation Loading...

BibTeX formatted citation

×
Data provided by:

Bookmark

BibSonomy logo Reddit logo
Bibliographic Tools

Bibliographic and Citation Tools

Bibliographic Explorer Toggle
Bibliographic Explorer (What is the Explorer?)
Connected Papers Toggle
Connected Papers (What is Connected Papers?)
Litmaps Toggle
Litmaps (What is Litmaps?)
scite.ai Toggle
scite Smart Citations (What are Smart Citations?)
Code, Data, Media

Code, Data and Media Associated with this Article

alphaXiv Toggle
alphaXiv (What is alphaXiv?)
Links to Code Toggle
CatalyzeX Code Finder for Papers (What is CatalyzeX?)
DagsHub Toggle
DagsHub (What is DagsHub?)
GotitPub Toggle
Gotit.pub (What is GotitPub?)
Huggingface Toggle
Hugging Face (What is Huggingface?)
Links to Code Toggle
Papers with Code (What is Papers with Code?)
ScienceCast Toggle
ScienceCast (What is ScienceCast?)
Demos

Demos

Replicate Toggle
Replicate (What is Replicate?)
Spaces Toggle
Hugging Face Spaces (What is Spaces?)
Spaces Toggle
TXYZ.AI (What is TXYZ.AI?)
Related Papers

Recommenders and Search Tools

Link to Influence Flower
Influence Flower (What are Influence Flowers?)
Core recommender toggle
CORE Recommender (What is CORE?)
IArxiv recommender toggle
IArxiv Recommender (What is IArxiv?)
About arXivLabs

arXivLabs: experimental projects with community collaborators

arXivLabs is a framework that allows collaborators to develop and share new arXiv features directly on our website.

Both individuals and organizations that work with arXivLabs have embraced and accepted our values of openness, community, excellence, and user data privacy. arXiv is committed to these values and only works with partners that adhere to them.

Have an idea for a project that will add value for arXiv's community? Learn more about arXivLabs.