root cause mitigation in AI training