Code for "MoME: Mixture-of-Masked-Experts for Efficient Multi-Task Recommendation", Jiahui Xu, Lu Sun, Dengji Zhao
Some implementations refers to the following repositories:
- L0_regularization: The implementation of "Learning Sparse Neural Networks through L0 regularization" by Christos Louizos, Max Welling & Diederik P. Kingma.
- stg: The implementation of "Feature Selection using Stochastic Gates" by Yamada, Yutaro and Lindenbaum, Ofir and Negahban, Sahand and Kluger, Yuval.
- MTReclib: MTReclib provides a PyTorch implementation of multi-task recommendation models and common datasets.