A TensorFlow Keras implementation of "Modeling Task Relationships in Multi-task Learning with Multi-gate Mixture-of-Experts" (KDD 2018)
This repo contains the implementation of Multi-gate Mixture-of-Experts model in TensorFlow Keras.
Here's the video explanation of the paper by the authors.
The repository includes:
The code is documented and designed to be extended relatively easy. If you plan on using this in your work, please consider citing this repository (BibTeX is included below) and also the paper.
requirements.txt
pip install -r requirements.txt
python census_income_demo.py
Contributions to this repository are welcome. Examples of things you can contribute:
Use this BibTeX to cite the repository:
@misc{keras_mmoe_2018,
title={Multi-gate Mixture-of-Experts model in Keras and TensorFlow},
author={Deng, Alvin},
year={2018},
publisher={Github},
journal={GitHub repository},
howpublished={\url{https://github.com/drawbridge/keras-mmoe}},
}
The code is built upon the work by Emin Orhan.