Tensor Product Attention Is All You Need
Conference on Neural Information Processing Systems (NeurIPS 2025 Spotlight)
Autonomous Data Selection with Zero-shot Generative Classifiers for Mathematical Texts
Findings of the Association for Computational Linguistics (ACL 2025 Findings)
Beyond Bradley-Terry Models: A General Preference Model for Language Model Alignment
International Conference on Machine Learning (ICML 2025)
Augmenting Math Word Problems via Iterative Question Composing
AAAI Conference on Artificial Intelligence (AAAI 2025)
Beyond Squared Error: Exploring Loss Design for Enhanced Training of Generative Flow Networks
International Conference on Learning Representations (ICLR 2025 Spotlight)
Information Flow in Self-Supervised Learning
International Conference on Machine Learning (ICML 2024)
Matrix Information Theory for Self-Supervised Learning
International Conference on Machine Learning (ICML 2024)
Cumulative Reasoning with Large Language Models
Transactions on Machine Learning Research (TMLR)
Contrastive Learning Is Spectral Clustering On Similarity Graph
International Conference on Learning Representations (ICLR 2024)
Trade-off Between Efficiency and Consistency for Removal-based Explanations
Conference on Neural Information Processing Systems (NeurIPS 2023)
(* denotes equal contribution, † denotes corresponding authors)