Paper ID: 2112.08538

Visualizing the Loss Landscape of Winning Lottery Tickets

Robert Bain

The underlying loss landscapes of deep neural networks have a great impact on their training, but they have mainly been studied theoretically due to computational constraints. This work vastly reduces the time required to compute such loss landscapes, and uses them to study winning lottery tickets found via iterative magnitude pruning. We also share results that contradict previously claimed correlations between certain loss landscape projection methods and model trainability and generalization error.

Submitted: Dec 16, 2021