← Search

Xiangxiang Xu

4 accepted papers

2026

Contrastive Predictive Coding Done Right for Mutual Information Estimation

ICLR 2026poster

The InfoNCE objective, originally introduced for contrastive representation learning, has become a popular choice for mutual information (MI) estimation, despite its indirect connection to MI. In this paper, we demonstrate why InfoNCE should not be regarded as a valid MI estimator, and we introduce…

Cited by 0SourceScholar
2026

Jet Expansions: Restructuring LLM Computation for Model Inspection

ICLR 2026poster

Large language models are becoming general knowledge engines for diverse applications. However, their computations are deeply entangled after training, resisting modularization which complicates interpretability, auditing, and long-term maintenance. We introduce Jet Expansions, a framework for expan…

Cited by 0SourceScholar
2024

Operator SVD with Neural Networks via Nested Low-Rank Approximation

ICML 2024poster

Computing eigenvalue decomposition (EVD) of a given linear operator, or finding its leading eigenvalues and eigenfunctions, is a fundamental task in many machine learning and scientific simulation problems. For high-dimensional eigenvalue problems, training neural networks to parameterize the eigenf…

2021

A Mathematical Framework for Quantifying Transferability in Multi-source Transfer Learning

NeurIPS 2021poster

Current transfer learning algorithm designs mainly focus on the similarities between source and target tasks, while the impacts of the sample sizes of these tasks are often not sufficiently addressed. This paper proposes a mathematical framework for quantifying the transferability in multi-source tr…

Cited by 37SourcePDFScholar