Environment Diversification with Multi-head Neural Network for Invariant Learning

Bo-Wei Huang, Keng-Te Liao, Chang-Sheng Kao +1

Neural networks are often trained with empirical risk minimization; however, it has been shown that a shift between training and testing distributions can cause unpredictable perfo…