Environment Diversification with Multi-head Neural Network for Invariant Learning