English

Farthest Greedy Path Sampling for Two-shot Recommender Search

Machine Learning 2023-11-01 v1 Information Retrieval

Abstract

Weight-sharing Neural Architecture Search (WS-NAS) provides an efficient mechanism for developing end-to-end deep recommender models. However, in complex search spaces, distinguishing between superior and inferior architectures (or paths) is challenging. This challenge is compounded by the limited coverage of the supernet and the co-adaptation of subnet weights, which restricts the exploration and exploitation capabilities inherent to weight-sharing mechanisms. To address these challenges, we introduce Farthest Greedy Path Sampling (FGPS), a new path sampling strategy that balances path quality and diversity. FGPS enhances path diversity to facilitate more comprehensive supernet exploration, while emphasizing path quality to ensure the effective identification and utilization of promising architectures. By incorporating FGPS into a Two-shot NAS (TS-NAS) framework, we derive high-performance architectures. Evaluations on three Click-Through Rate (CTR) prediction benchmarks demonstrate that our approach consistently achieves superior results, outperforming both manually designed and most NAS-based models.

Keywords

Cite

@article{arxiv.2310.20705,
  title  = {Farthest Greedy Path Sampling for Two-shot Recommender Search},
  author = {Yufan Cao and Tunhou Zhang and Wei Wen and Feng Yan and Hai Li and Yiran Chen},
  journal= {arXiv preprint arXiv:2310.20705},
  year   = {2023}
}

Comments

9 pages, 5 figures

R2 v1 2026-06-28T13:07:46.295Z