Publications
Robust Machine Learning
Conference Paper
Cross-Entropy Is All You Need To Invert the Data Generating Process
Reizinger, P., Bizeul, A., Juhos, A., Vogt, J. E., Balestriero, R., Brendel, W., Klindt, D.
In January 2025 (Published)
OpenReview
BibTeX
Empirical Inference
Article
Flow Matching for Atmospheric Retrieval of Exoplanets: Where Reliability meets Adaptive Noise Levels
Gebhard, T. D., Wildberger, J., Dax, M., Kofler, A., Angerhausen, D., Quanz, S., Schölkopf, B.
Astronomy \& Astrophysics, 693, January 2025 (Published)
DOI
URL
BibTeX
Robust Machine Learning
Conference Paper
Identifiable Exchangeable Mechanisms for Causal Structure and Representation Learning
Reizinger, P., Guo, S., Huszár, F., Schölkopf, B., Brendel, W.
In January 2025 (Published)
OpenReview
BibTeX
Robust Machine Learning
Conference Paper
In Search of Forgotten Domain Generalization
Mayilvahanan, P., Zimmermann, R. S., Wiedemer, T., Rusak, E., Juhos, A., Bethge, M., Brendel, W.
In January 2025 (Published)
OpenReview
BibTeX
Robust Machine Learning
Conference Paper
Interaction Asymmetry: A General Principle for Learning Composable Abstractions
Brady, J., von Kügelgen, J., Lachapelle, S., Buchholz, S., Kipf, T., Brendel, W.
In January 2025 (Published)
OpenReview
BibTeX
Empirical Inference
Ph.D. Thesis
Machine Learning Meets Exoplanet Science: Methodical contributions to direct imaging and atmospheric retrieval
Gebhard, T.
ETH Zurich, Switzerland, January 2025 (Published)
BibTeX
Deep Models and Optimization
Conference Paper
Using Shapley interactions to understand how models use structure
Divyansh Singhvi, D. M. A. E. R. J. I. P. N. S.
In Proceedings ACL, 1-20, Vienna Center, Association for Computational Linguistics (ACL 2025), 2025 (Accepted)
DOI
URL
BibTeX
Deep Models and Optimization
Article
An accelerated lyapunov function for Polyak’s Heavy-ball on convex quadratics
Orvieto, A.
Optimization Letters, 19:307-328, 2025 (Published)
DOI
URL
BibTeX
Empirical Inference
Technical Report
International AI Safety Report
Bengio, Y., Mindermann, S., Privitera, D., Besiroglu, T., Bommasani, R., Casper, S., Choi, Y., Fox, P., Garfinkel, B., Goldfarb, D., et al.
(DSIT 2025/001), 2025 (Published)
URL
BibTeX
Empirical Inference
Book Chapter
Natural Language Processing
Jin, Z., Mihalcea, R., Schölkopf, B.
In Elgar Encyclopedia of Political Communication, (Editors: Nai, A. and Grömping, M. and Wirz, D.), Edward Elgar Publishing, 2025 (Published)
PDF
URL
BibTeX
Safety- and Efficiency- aligned Learning
Conference Paper
Efficiently Dispatching Flash Attention For Partially Filled Attention Masks
Sharma, A., Geiping, J.
In ENSLP NeurIPS Workshop 2024, ENSLP NeurIPS Workshop 2024, ENSLP NeurIPS Workshop, December 2024 (Published)
URL
BibTeX
Empirical Inference
Conference Paper
From Causal to Concept-Based Representation Learning
Rajendran*, G., Buchholz*, S., Aragam, B., Schölkopf, B., Ravikumar, P. K.
Advances in Neural Information Processing Systems 37 (NeurIPS 2024), 37:101250-101296, (Editors: A. Globerson and L. Mackey and D. Belgrave and A. Fan and U. Paquet and J. Tomczak and C. Zhang), Curran Associates, Inc., 38th Annual Conference on Neural Information Processing Systems, December 2024 (Published)
URL
BibTeX
Empirical Inference
Conference Paper
Learning Partitions from Context
Buchholz, S.
Advances in Neural Information Processing Systems 37 (NeurIPS 2024), 37:140066-140112, (Editors: A. Globerson and L. Mackey and D. Belgrave and A. Fan and U. Paquet and J. Tomczak and C. Zhang), Curran Associates, Inc., 38th Annual Conference on Neural Information Processing Systems, December 2024 (Published)
URL
BibTeX
Empirical Inference
Conference Paper
Metacognitive Capabilities of LLMs: An Exploration in Mathematical Problem Solving
Didolkar, A. R., Goyal, A., Ke, N. R., Guo, S., Valko, M., Lillicrap, T. P., Rezende, D. J., Bengio, Y., Mozer, M. C., Arora, S.
Advances in Neural Information Processing Systems 37 (NeurIPS 2024), 37:19783-19812, (Editors: A. Globerson and L. Mackey and D. Belgrave and A. Fan and U. Paquet and J. Tomczak and C. Zhang), Curran Associates, Inc., 38th Annual Conference on Neural Information Processing Systems, December 2024 (Published)
URL
BibTeX
Empirical Inference
Conference Paper
A Generative Model of Symmetry Transformations
Allingham, J. U., Mlodozeniec, B. K., Padhy, S., Antorán, J., Krueger, D., Turner, R. E., Nalisnick, E., Hernández-Lobato, J. M.
Advances in Neural Information Processing Systems 37 (NeurIPS 2024), 37:91091-91130, (Editors: A. Globerson and L. Mackey and D. Belgrave and A. Fan and U. Paquet and J. Tomczak and C. Zhang), Curran Associates, Inc., 38th Annual Conference on Neural Information Processing Systems, December 2024 (Published)
URL
BibTeX
Empirical Inference
Article
A Randomized Controlled Trial on Anonymizing Reviewers to Each Other in Peer Review Discussions
Rastogi, C., Song, X., Jin, Z., Stelmakh, I., Daumé III, H., Zhang, K., Shah, N. B.
PLOS ONE, 19(12), Public Library of Science, December 2024 (Published)
DOI
URL
BibTeX
Social Foundations of Computation
Algorithms and Society
Conference Paper
Algorithmic Collective Action in Recommender Systems: Promoting Songs by Reordering Playlists
Baumann, J., Mendler-Dünner, C.
Advances in Neural Information Processing Systems 37 (NeurIPS 2024) , The Thirty-Eighth Annual Conference on Neural Information Processing Systems (NeurIPS), December 2024 (Published)
arXiv
URL
BibTeX
Empirical Inference
Conference Paper
Alien Recombination: Exploring Concept Blends Beyond Human Cognitive Availability in Visual Art
Hernandez, A., Brinkmann, L., Serna, I., Rahaman, N., Alhaija, H. A., Yakura, H., Sola, M. C., Schölkopf, B., Rahwan, I.
NeurIPS 2024 Workshop on Creativity and Generative AI, December 2024 (Published)
arXiv
BibTeX
Social Foundations of Computation
Algorithms and Society
Conference Paper
An Engine Not a Camera: Measuring Performative Power of Online Search
Mendler-Dünner, C., Carovano, G., Hardt, M.
Advances in Neural Information Processing Systems 37 (NeurIPS 2024), The Thirty-Eighth Annual Conference on Neural Information Processing Systems (NeurIPS), December 2024 (Published)
ArXiv
URL
BibTeX
Empirical Inference
Conference Paper
Causal vs. Anticausal merging of predictors
Garrido Mejia, S., Blöbaum, P., Schölkopf, B., Janzing, D.
Advances in Neural Information Processing Systems 37 (NeurIPS 2024) , 37:1402-1427, (Editors: A. Globerson and L. Mackey and D. Belgrave and A. Fan and U. Paquet and J. Tomczak and C. Zhang), Curran Associates, Inc., 38th Annual Conference on Neural Information Processing Systems, December 2024 (Published)
URL
BibTeX
Empirical Inference
Ph.D. Thesis
Causality for Natural Language Processing
Jin, Z.
University of Tübingen, Germany, December 2024, (ELLIS PhD student program) (Published)
URL
BibTeX
Empirical Inference
Conference Paper
Causally Testing Gender Bias in LLMs: A Case Study on Occupational Bias
Chen*, Y., Vethavikashini*, C. R., Mattern*, J., Mihalcea, R., Jin, Z.
NeurIPS 2024 Workshop on Causality and Language Models (CaLM), December 2024, *equal contribution (Published)
DOI
URL
BibTeX
Empirical Inference
Conference Paper
Cooperate or Collapse: Emergence of Sustainability in a Society of LLM Agents
Piatti*, G., Jin*, Z., Kleiman-Weiner*, M., Schölkopf, B., Sachan, M., Mihalcea, R.
Advances in Neural Information Processing Systems 37 (NeurIPS 2024), 37:111715-111759, (Editors: A. Globerson and L. Mackey and D. Belgrave and A. Fan and U. Paquet and J. Tomczak and C. Zhang), Curran Associates, Inc., 38th Annual Conference on Neural Information Processing Systems, December 2024, *equal contribution (Published)
arXiv
URL
BibTeX
Empirical Inference
Conference Paper
Do Finetti: On Causal Effects for Exchangeable Data
Guo, S., Zhang, C., Muhan, K., Huszár*, F., Schölkopf*, B.
Advances in Neural Information Processing Systems 37 (NeurIPS 2024), 37:127317-127345, (Editors: A. Globerson and L. Mackey and D. Belgrave and A. Fan and U. Paquet and J. Tomczak and C. Zhang), Curran Associates, Inc., 38th Annual Conference on Neural Information Processing Systems, December 2024, *equal supervision (Published)
URL
BibTeX
Social Foundations of Computation
Algorithms and Society
Conference Paper
Evaluating Language Models as Risk Scores
Cruz, A. F., Hardt, M., Mendler-Dünner, C.
Advances in Neural Information Processing Systems 37 (NeurIPS 2024), The Thirty-Eighth Annual Conference on Neural Information Processing Systems (NeurIPS), December 2024 (Published)
ArXiv
Code
URL
BibTeX
Empirical Inference
Conference Paper
Improving Linear System Solvers for Hyperparameter Optimisation in Iterative Gaussian Processes
Lin, J. A., Padhy, S., Mlodozeniec, B. K., Antorán, J., Hernández-Lobato, J. M.
Advances in Neural Information Processing Systems 37 (NeurIPS 2024) , 37:15460-15496, (Editors: A. Globerson and L. Mackey and D. Belgrave and A. Fan and U. Paquet and J. Tomczak and C. Zhang), Curran Associates, Inc., 38th Annual Conference on Neural Information Processing Systems, December 2024 (Published)
URL
BibTeX
Empirical Inference
Conference Paper
Inferring stochastic low-rank recurrent neural networks from neural data
Pals, M., Sağtekin, A. E., Pei, F., Gloeckler, M., Macke, J.
Advances in Neural Information Processing Systems 37 (NeurIPS 2024) , 37:18225-18264, (Editors: A. Globerson and L. Mackey and D. Belgrave and A. Fan and U. Paquet and J. Tomczak and C. Zhang), Curran Associates, Inc., 38th Annual Conference on Neural Information Processing Systems, December 2024 (Published)
URL
BibTeX
Empirical Inference
Conference Paper
Latent Diffusion for Neural Spiking Data
Kapoor, J., Schulz, A., Vetter, J., Pei, F., Gao, R., Macke, J. H.
Advances in Neural Information Processing Systems 37 (NeurIPS 2024), 37:118119-118154, (Editors: A. Globerson and L. Mackey and D. Belgrave and A. Fan and U. Paquet and J. Tomczak and C. Zhang), Curran Associates, Inc., 38th Annual Conference on Neural Information Processing Systems, December 2024 (Published)
URL
BibTeX
Empirical Inference
Conference Paper
Limits of Transformer Language Models on Learning to Compose Algorithms
Thomm, J., Camposampiero, G., Terzic, A., Hersche, M., Schölkopf, B., Rahimi, A.
Advances in Neural Information Processing Systems 37 (NeurIPS 2024), 37:7631-7674, (Editors: A. Globerson and L. Mackey and D. Belgrave and A. Fan and U. Paquet and J. Tomczak and C. Zhang), Curran Associates, Inc., 38th Annual Conference on Neural Information Processing Systems, December 2024 (Published)
arXiv
URL
BibTeX
Empirical Inference
Conference Paper
Neural Characteristic Activation Analysis and Geometric Parameterization for ReLU Networks
Chen, W., Ge, H.
Advances in Neural Information Processing Systems 37 (NeurIPS 2024) , 37:97562-97586, (Editors: A. Globerson and L. Mackey and D. Belgrave and A. Fan and U. Paquet and J. Tomczak and C. Zhang), Curran Associates, Inc., 38th Annual Conference on Neural Information Processing Systems, December 2024 (Published)
URL
BibTeX
Empirical Inference
Conference Paper
On Affine Homotopy between Language Encoders
Chan, R., Bourmasmoud, R., Svete, A., Ren, Y., Guo, Q., Jin, Z., Ravfogel, S., Sachan, M., Schölkopf, B., El-Assady, M., et al.
Advances in Neural Information Processing Systems 37 (NeurIPS 2024), 37:73337-73365, (Editors: A. Globerson and L. Mackey and D. Belgrave and A. Fan and U. Paquet and J. Tomczak and C. Zhang), Curran Associates, Inc., 38th Annual Conference on Neural Information Processing Systems, December 2024 (Published)
URL
BibTeX
Social Foundations of Computation
Algorithms and Society
Conference Paper
Questioning the Survey Responses of Large Language Models
Dominguez-Olmedo, R., Hardt, M., Mendler-Dünner, C.
Advances in Neural Information Processing Systems 37 (NeurIPS 2024), The Thirty-Eighth Annual Conference on Neural Information Processing Systems (NeurIPS), Oral, December 2024 (Published)
ArXiv
URL
BibTeX
Empirical Inference
Conference Paper
Shaving Weights with Occam’s Razor: Bayesian Sparsification for Neural Networks using the Marginal Likelihood
Dhahri, R., Immer, A., Charpentier, B., Günnemann, S., Fortuin, V.
Advances in Neural Information Processing Systems 37 (NeurIPS 2024), 37:24959-24989, (Editors: A. Globerson and L. Mackey and D. Belgrave and A. Fan and U. Paquet and J. Tomczak and C. Zhang), Curran Associates, Inc., 38th Annual Conference on Neural Information Processing Systems, December 2024 (Published)
URL
BibTeX
Empirical Inference
Conference Paper
Sourcerer: Sample-based Maximum Entropy Source Distribution Estimation
Vetter, J., Moss, G., Schröder, C., Gao, R., Macke, J. H.
Advances in Neural Information Processing Systems 37 (NeurIPS 2024), 37:88772-88806, (Editors: A. Globerson and L. Mackey and D. Belgrave and A. Fan and U. Paquet and J. Tomczak and C. Zhang), Curran Associates, Inc., 38th Annual Conference on Neural Information Processing Systems, December 2024 (Published)
URL
BibTeX
Empirical Inference
Conference Paper
Theoretical Characterisation of the Gauss Newton Conditioning in Neural Networks
Zhao*, J., Singh*, S. P., Lucchi, A.
Advances in Neural Information Processing Systems 37 (NeurIPS 2024), 37:114965-115000, (Editors: A. Globerson and L. Mackey and D. Belgrave and A. Fan and U. Paquet and J. Tomczak and C. Zhang), Curran Associates, Inc., 38th Annual Conference on Neural Information Processing Systems, December 2024, *equal contribution (Published)
URL
BibTeX
Empirical Inference
Conference Paper
What Makes and Breaks Safety Fine-tuning? A Mechanistic Study
Jain, S., Lubana, E. S., Oksuz, K., Joy, T., Torr, P., Sanyal, A., Dokania, P. K.
Advances in Neural Information Processing Systems 37 (NeurIPS 2024), 37:93406-93478, (Editors: A. Globerson and L. Mackey and D. Belgrave and A. Fan and U. Paquet and J. Tomczak and C. Zhang), Curran Associates, Inc., 38th Annual Conference on Neural Information Processing Systems, December 2024 (Published)
URL
BibTeX
Deep Models and Optimization
Conference Paper
Adaptive Methods through the Lens of SDEs: Theoretical Insights on the Role of Noise
Monzio Compagnoni, E., Liu, T., Islamov, R., Proske, F. N., Orvieto, A., Lucchi, A.
In The Thirteenth International Conference on Learning Representations, ICLR 2025, The Thirteenth International Conference on Learning Representations, November 2024 (Accepted)
BibTeX
Safety- and Efficiency- aligned Learning
Conference Paper
Democratizing AI: Open-source Scalable LLM Training on GPU-based Supercomputers
Singh, S., Singhania, P., Ranjan, A., Kirchenbauer, J., Geiping, J., Wen, Y., Jain, N., Hans, A., Shu, M., Tomar, A., et al.
International Conference for High Performance Computing, Networking, Storage and Analysis SC (SC24), 36-49, Supercomputing, IEEE Digital Library, Atlanta, GA, International Conference for High Performance Computing, November 2024 (Published)
DOI
URL
BibTeX
Robust Machine Learning
Article
Interaction Asymmetry: A General Principle for Learning Composable Abstractions
Brady, J., von Kügelgen, J., Lachapelle, S., Buchholz, S., Kipf, T., Brendel, W.
November 2024 (Submitted)
BibTeX
Haptic Intelligence
Autonomous Learning
Empirical Inference
Miscellaneous
Demonstration: Minsight - A Soft Vision-Based Tactile Sensor for Robotic Fingertips
Andrussow, I., Sun, H., Martius, G., Kuchenbecker, K. J.
Hands-on demonstration presented at the Conference on Robot Learning (CoRL), Munich, Germany, November 2024 (Published)
BibTeX
Empirical Inference
Conference Paper
Diffusion-based learning of contact plans for agile locomotion
Dh’Edin, V., Ravi, A. K. C., Jordana, A., Zhu, H., Meduri, A., Righetti, L., Schölkopf, B., Khadiv, M.
IEEE-RAS 23rd International Conference on Humanoid Robots (Humanoids), 637-644, IEEE, November 2024 (Published)
DOI
URL
BibTeX
Empirical Inference
Conference Paper
Do LLMs Think Fast and Slow? A Causal Study on Sentiment Analysis
Lyu*, Z., Jin*, Z., Gonzalez, F., Mihalcea, R., Schölkopf, B., Sachan, M.
Findings of the Association for Computational Linguistics: EMNLP, 9353-9372, (Editors: Yaser Al-Onaizan and Mohit Bansal and Yun-Nung Chen), Association for Computational Linguistics, November 2024, *equal contribution (Published)
DOI
URL
BibTeX
Empirical Inference
Conference Paper
Exploring the Jungle of Bias: Political Bias Attribution in Language Models via Dependency Analysis
Jenny*, D. F., Billeter*, Y., Schölkopf, B., Jin, Z.
Proceedings of the Third Workshop on NLP for Positive Impact, 152-178, (Editors: Dementieva, Daryna and Ignat, Oana and Jin, Zhijing and Mihalcea, Rada and Piatti, Giorgio and Tetreault, Joel and Wilson, Steven and Zhao, Jieyu), Association for Computational Linguistics, November 2024, *equal contribution (Published)
URL
BibTeX
Empirical Inference
Conference Paper
Implicit Personalization in Language Models: A Systematic Study
Jin, Z., Heil, N., Liu, J., Dhuliawala, S., Qi, Y., Schölkopf, B., Mihalcea, R., Sachan, M.
Findings of the Association for Computational Linguistics: EMNLP, 12309-12325, (Editors: Yaser Al-Onaizan and Mohit Bansal and Yun-Nung Chen), Association for Computational Linguistics, November 2024 (Published)
DOI
URL
BibTeX
Empirical Inference
Ph.D. Thesis
On Principled Modeling of Inductive Bias in Machine Learning
Liu, W.
University of Cambridge, UK, Cambridge, November 2024, (Cambridge-Tübingen-Fellowship-Program, ELLIS PhD student program) (Published)
BibTeX
Empirical Inference
Conference Paper
The Odyssey of Commonsense Causality: From Foundational Benchmarks to Cutting-Edge Reasoning
Cui, S., Jin, Z., Schölkopf, B., Faltings, B.
Proceedings of the 2024 Conference on Empirical Methods in Natural Language Processing (EMNLP), 16722-16763, (Editors: Al-Onaizan, Yaser and Bansal, Mohit and Chen, Yun-Nung), Association for Computational Linguistics, November 2024 (Published)
URL
BibTeX
Empirical Inference
Conference Paper
RP1M: A Large-Scale Motion Dataset for Piano Playing with Bi-Manual Dexterous Robot Hands
Zhao*, Y., Chen*, L., Schneider, J., Gao, Q., Kannala, J., Schölkopf, B., Pajarinen, J., Büchler, D.
Proceedings of the 8th Annual Conference on Robot Learning (CoRL), 270:5184-5203, Proceedings of Machine Learning Research, (Editors: Agrawal, Pulkit and Kroemer, Oliver and Burgard, Wolfram), PMLR, Conference on Robot Learning, November 2024, *equal contribution (Published)
URL
BibTeX
Deep Models and Optimization
Article
NIMBA: Towards Robust and Principled Processing of Point Clouds With SSMs
Köprücü, N., Okpekpe, D., Orvieto, A.
October 2024 (In preparation)
BibTeX
Safety- and Efficiency- aligned Learning
Technical Report
A Realistic Threat Model for Large Language Model Jailbreaks
Boreiko, V., Panfilov, A., Hein, M., Geiping, J.
October 2024 (Submitted)
URL
BibTeX
Deep Models and Optimization
Conference Paper
Loss Landscape Characterization of Neural Networks without Over-Parametrization
Islamov, R., Ajroldi, N., Orvieto, A., Lucchi, A.
In Proceedings of the Thirty-Eighth Annual Conference on Neural Information Processing Systems, Thirty-Eighth Annual Conference on Neural Information Processing Systems, October 2024 (Published)
URL
BibTeX