Timezone: »

Towards Modular Machine Learning Pipelines
Aditya Modi · JIVAT NEET KAUR · Maggie Makar · Pavan Mallapragada · Amit Sharma · Emre Kiciman · Adith Swaminathan
Event URL: https://openreview.net/forum?id=SiSID2Wo6j »

Pipelines of Machine Learning (ML) components are a popular and effective approach to divide and conquer many business-critical problems. A pipeline architecture implies a specific division of the overall problem, however current ML training approaches do not enforce this implied division. Consequently ML components can become coupled to one another after they are trained, which causes insidious effects. For instance, even when one coupled ML component in a pipeline is improved in isolation, the end-to-end pipeline performancecan degrade. In this paper, we develop a conceptual framework to study ML coupling in pipelines and design new modularity regularizersthat can eliminate coupling during ML training. We show that the resulting ML pipelines become modular (i.e., their components can be trained independently of one another) and discuss the tradeoffs of our approach versus existing approaches to pipeline optimization.

Author Information

Aditya Modi (Microsoft)
JIVAT NEET KAUR (Microsoft Research, India)
Maggie Makar (University of Michigan)
Pavan Mallapragada
Amit Sharma (Microsoft Research)
Emre Kiciman (Microsoft Research)
Adith Swaminathan (Microsoft Research)

More from the Same Authors