Section 4.4: Robust Model Training