SWAD: Domain Generalization by Seeking Flat Minima

[tkuri]

論文概要

平坦な最小値(Flat minima)を求めることでドメイン一般化におけるギャップが小さくなることを理論的に示しFlat minimaを求めるためのシンプルで効率的なSWADを提案。またフラット性を考慮しないMixupやCutMixはドメイン外の汎化性能を保証できない事を理論・実践的に示した。

https://arxiv.org/abs/2102.08604

Code

https://github.com/khanrc/swad