Yue Song

I am a Computing & Mathematical Sciences postdoctoral research associate at Caltech, supervised by Yisong Yue, Pietro Perona, and Max Welling. I pursued doctoral studies under European Laboratory for Learning and Intelligent Systems (ELLIS), where I was affiliated with Multimedia and Human Understanding Group (MHUG) at University of Trento, Italy and Amsterdam Machine Learning Lab (AMLab) at University of Amsterdam, the Netherlands, advised by Nicu Sebe and Max Welling.

I research structured representation learning — I am devoted to leveraging beneficial inductive biases from scientific disciplines such as math, physics, and neuroscience to improve and explain machine learning models. My current research is not task-oriented; I do not focus on a particular ML task. Instead, I am interested in developing structured methods and finding their appropriate usage in the wide application domain.

The specific deep learning fields I have worked on include high-order representation learning, decorrelated representation learning, equivariant representation learning, disentangled representation learning, and detecting/handling distribution shifts. On a theoretical aspect, the developed methodologies involve numerical and statistical matrix analysis, computational methods of matrix functions/decompositions, physics-inspired deep learning, variational inference, and matrix manifold learning.

Prior to my Ph.D. studies, I received the B.Sc. cum laude from KU Leuven, Belgium and the joint M.Sc. summa cum laude from University of Trento, Italy and KTH Royal Institute of Technology, Sweden. Besides the technical master's degree, I received an Innovation & Entrepreneurship minor degree from European Institute of Innovation and Technology (EIT Digital).

News

2025/04: Serve as Area Chair for NeurIPS 2025.
2024/02: Give CVPR and ECCV tutorials on disentangled&equivariant representation learning.

Selected Publications

	Unsupervised Representation Learning from Sparse Transformation Analysis Yue Song, T. Anderson Keller, Yisong Yue, Pietro Perona, Max Welling submitted to IEEE Transactions on Pattern Analysis and Machine Intelligence (T-PAMI), 2024 paper / code / bibtex
	Navigating Chemical Space with Latent Flows Guanghao Wei, Yining Huang, Chenru Duan, Yue Song♦, Yuanqi Du♦ Advances in Neural Information Processing Systems (NeurIPS), 2024 paper / code / bibtex (♦ denotes equal supervision)
	RMLR: Extending Multinomial Logistic Regression into General Geometries Ziheng Chen, Yue Song♠, Rui Wang, Xiaojun Wu, Nicu Sebe Advances in Neural Information Processing Systems (NeurIPS), 2024 paper / code / bibtex (♠ denotes the corresponding author)
	A Lie Group Approach to Riemannian Batch Normalization Ziheng Chen, Yue Song♠, Yunmei Liu, Nicu Sebe International Conference on Learning Representations (ICLR), 2024 paper / code / bibtex (♠ denotes the corresponding author)
	Flow Factorized Representation Learning Yue Song, T. Anderson Keller, Nicu Sebe, Max Welling Advances in Neural Information Processing Systems (NeurIPS), 2023 paper / code / bibtex
	Householder Projector for Unsupervised Latent Semantics Discovery Yue Song, Jichao Zhang, Nicu Sebe, Wei Wang International Conference on Computer Vision (ICCV), 2023 paper / code / bibtex
	Latent Traversals in Generative Models as Potential Flows Yue Song, T. Anderson Keller, Nicu Sebe, Max Welling International Conference on Machine Learning (ICML), 2023 paper / code / bibtex
	Masked Jigsaw Puzzle: A Versatile Position Embedding for Vision Transformers Bin Ren♣, Yahui Liu♣, Yue Song, Wei Bi, Rita Cucchiara, Nicu Sebe, Wei Wang International Conference on Computer Vision and Pattern Recognition (CVPR), 2023 paper / code / bibtex (♣ denotes co-first author)
	Orthogonal SVD Covariance Conditioning and Latent Disentanglement Yue Song, Nicu Sebe, Wei Wang IEEE Transactions on Pattern Analysis and Machine Intelligence (T-PAMI), 2022 paper / code / bibtex
	Fast Differentiable Matrix Square Root and Inverse Square Root Yue Song, Nicu Sebe, Wei Wang IEEE Transactions on Pattern Analysis and Machine Intelligence (T-PAMI), 2022 paper / code / bibtex
	RankFeat: Rank-1 Feature Removal for Out-of-distribution Detection Yue Song, Nicu Sebe, Wei Wang Advances in Neural Information Processing Systems (NeurIPS), 2022 paper / code / slides / bibtex
	Batch-efficient EigenDecomposition for Small and Medium Matrices Yue Song, Nicu Sebe, Wei Wang European Conference on Computer Vision (ECCV), 2022 paper / code / slides / bibtex
	Improving Covariance Conditioning of the SVD Meta-layer by Orthogonality Yue Song, Nicu Sebe, Wei Wang European Conference on Computer Vision (ECCV), 2022 paper / code / slides / bibtex
	On the Eigenvalues of Global Covariance Pooling for Fine-grained Visual Recognition Yue Song, Nicu Sebe, Wei Wang IEEE Transactions on Pattern Analysis and Machine Intelligence (T-PAMI), 2022 paper / code / bibtex
	Fast Differentiable Matrix Square Root Yue Song, Nicu Sebe, Wei Wang International Conference on Learning Representations (ICLR), 2022, (Top 2.9%) paper / code / slides / bibtex
	Why Approximate Matrix Square Root Outperforms Accurate SVD in Global Covariance Pooling? Yue Song, Nicu Sebe, Wei Wang International Conference on Computer Vision (ICCV), 2021 paper / supplementary / code / slides / poster / bibtex