The Vanishing Decision Boundary Complexity and the Strong First Component

November 25, 2022 · Entered Twilight · 🏛 arXiv.org

Repo contents: .gitignore, README.MD, cifar10_with_id.py, decision_boundary_evolution_video.py, extract_test_objects_classwise.py, generate_cctm.py, generate_fig7.py, generate_mean_and_variance.py, main.py, measure_cross_class_distances.py, models, output_space.py, plot_boundary_convergence_after_compare_final_class_var.py, plot_cat_dog_in_PCA2.py, plot_fig1_fig2_fig3_fig6_fig9_fig10.py, plot_fig4_fig5_fig8.py, plot_final_sigma1.py, plot_predecessor_boundaries_nearest.py, study_final_model_low_confidence_samples.py, utils.py

Authors Hengshuai Yao arXiv ID 2211.16209 Category cs.LG: Machine Learning Cross-listed cs.AI Citations 0 Venue arXiv.org Repository https://github.com/hengshu1/decision_boundary_github ⭐ 1 Last Checked 3 months ago

Abstract

We show that unlike machine learning classifiers, there are no complex boundary structures in the decision boundaries for well-trained deep models. However, we found that the complicated structures do appear in training but they vanish shortly after shaping. This is a pessimistic news if one seeks to capture different levels of complexity in the decision boundary for understanding generalization, which works well in machine learning. Nonetheless, we found that the decision boundaries of predecessor models on the training data are reflective of the final model's generalization. We show how to use the predecessor decision boundaries for studying the generalization of deep models. We have three major findings. One is on the strength of the first principle component of deep models, another about the singularity of optimizers, and the other on the effects of the skip connections in ResNets. Code is at https://github.com/hengshu1/decision_boundary_github.