FlatNAS: Optimizing Neural Architecture for Out-of-Distribution Robustness

Neural Architecture Search (NAS) explores flat regions in loss landscapes to optimize NN architectures for out-of-distribution (OOD) robustness.