A Unified Perspective on Regularization and Perturbation in Differentiable Subset Selection

Xiangqian Sun, Cheuk Hang Leung, Yijun LI, Qi Wu*

*Corresponding author for this work

Research output: Chapters, Conference Papers, Creative and Literary WorksRGC 32 - Refereed conference paper (with host publication)peer-review

1 Citation (Scopus)

Abstract

Subset selection, i.e., finding a bunch of items from a collection to achieve specific goals, has wide applications in information retrieval, statistics, and machine learning. To implement an end-to-end learning framework, different relaxed differentiable operators of subset selection are proposed. Most existing work relies on either the regularization method or the perturbation method. In this work, we provide a probabilistic interpretation for regularization relaxation and unify two schemes. Besides, we build some concrete examples to show the generic connection between these two relaxations. Finally, we evaluate the perturbed selector as well as the regularized selector on two tasks: the maximum entropy sampling problem and the feature selection problem. The experimental results show that these two methods can achieve competitive performance against other benchmarks. © 2023 by the author(s).
Original languageEnglish
Title of host publicationProceedings of the 26th International Conference on Artificial Intelligence and Statistics (AISTATS) 2023
EditorsFrancisco Ruiz, Jennifer Dy, Jan-Willem van de Meent
PublisherPMLR
Pages4629-4642
Volume206
Publication statusPublished - Apr 2023
Event26th International Conference on Artificial Intelligence and Statistics (AISTATS 2023) - Palau de Congressos, Valencia, Spain
Duration: 25 Apr 202327 Apr 2023
http://aistats.org/aistats2023/#:~:text=The%2026th%20International%20Conference%20on,as%20an%20in%2Dperson%20event.

Publication series

NameProceedings of Machine Learning Research
ISSN (Print)2640-3498

Conference

Conference26th International Conference on Artificial Intelligence and Statistics (AISTATS 2023)
PlaceSpain
CityValencia
Period25/04/2327/04/23
Internet address

Funding

Qi Wu acknowledges the support from The CityU-JD Digits Joint Laboratory in Financial Technology and Engineering; The Hong Kong Research Grants Council [General Research Fund 14206117, 11219420, and 11200219]; The CityU SRG-Fd fund 7005300, and The HK Institute of Data Science. The work described in this paper was partially supported by the InnoHK initiative, The Government of the HKSAR, and the Laboratory for AI-Powered Financial Technologies

RGC Funding Information

  • RGC-funded

Fingerprint

Dive into the research topics of 'A Unified Perspective on Regularization and Perturbation in Differentiable Subset Selection'. Together they form a unique fingerprint.

Cite this