Publications | Rational Intelligence Lab

(*,† indicate equal contributions)

2025

Causality

When Shift Happens - Confounding Is to Blame

Abbavaram Gowtham Reddy, Celia Rubio-Madrigal, Rebekka Burkholz, Krikamol Muandet

preprint, 2025

DOI Bib PDF

@article{reddy2025when,
  title = {When Shift Happens - Confounding Is to Blame},
  author = {Reddy, Abbavaram Gowtham and Rubio-Madrigal, Celia and Burkholz, Rebekka and Muandet, Krikamol},
  journal = {preprint},
  year = {2025},
  doi = {10.48550/arXiv.2505.21422},
}

Probability Theory

Integral Imprecise Probability Metrics

Siu Lun Chau, Michele Caprio, Krikamol Muandet

preprint, 2025

DOI Bib PDF

@article{chau2025imprecise,
  title = {Integral Imprecise Probability Metrics},
  author = {Chau, Siu Lun and Caprio, Michele and Muandet, Krikamol},
  journal = {preprint},
  year = {2025},
  doi = {10.48550/arXiv.2505.16156},
}

Explainability

Computing Exact Shapley Values in Polynomial Time for Product-Kernel Methods

Majid Mohammadi, Siu Lun Chau, Krikamol Muandet

preprint, 2025

DOI Bib PDF

@article{majid2025computing,
  title = {Computing Exact Shapley Values in Polynomial Time for Product-Kernel Methods},
  author = {Mohammadi, Majid and Chau, Siu Lun and Muandet, Krikamol},
  journal = {preprint},
  year = {2025},
  doi = {10.48550/arXiv.2505.16516},
}

Regulation

Explanation Design in Strategic Learning: Sufficient Explanations that Induce Non-harmful Responses

Kiet QH Vo, Siu Lun Chau, Masahiro Kato, Yixin Wang, Krikamol Muandet

preprint, 2025

DOI Bib PDF Code

@article{vo2025strategic,
  title = {Explanation Design in Strategic Learning: Sufficient Explanations that Induce Non-harmful Responses},
  author = {Vo, Kiet QH and Chau, Siu Lun and Kato, Masahiro and Wang, Yixin and Muandet, Krikamol},
  journal = {preprint},
  year = {2025},
  doi = {10.48550/arXiv.2502.04058},
}

Probability Theory

Truthful Elicitation of Imprecise Forecasts

Anurag Singh, Siu Lun Chau, Krikamol Muandet

In Uncertainty in Aritificial Intelligence (UAI), 2025

Oral DOI Bib PDF Code Poster

Oral Presentation at UAI 2025

@inproceedings{singh2025truthful,
  title = {Truthful Elicitation of Imprecise Forecasts},
  author = {Singh, Anurag and Chau, Siu Lun and Muandet, Krikamol},
  booktitle = {Uncertainty in Aritificial Intelligence (UAI)},
  year = {2025},
  publisher = {PMLR},
  doi = {10.48550/arXiv.2503.16395},
}

Kernels

Kernel Quantile Embeddings and Associated Probability Metrics

Masha Naslidnyk, Siu Lun Chau, F-X Briol, Krikamol Muandet

In International Conference on Machine Learning (ICML), 2025

Bib PDF

@inproceedings{naslidnyk2025kqe,
  title = {Kernel Quantile Embeddings and Associated Probability Metrics},
  author = {Naslidnyk, Masha and Chau, Siu Lun and Briol, F-X and Muandet, Krikamol},
  booktitle = {International Conference on Machine Learning (ICML)},
  year = {2025},
  publisher = {PMLR},
}

Prediction

Teaching Transformers Causal Reasoning through Axiomatic Training

Aniket Vashishtha, Abhinav Kumar, Atharva Pandey, Abbavaram Gowtham Reddy, Kabir Ahuja, Vineeth N Balasubramanian, Amit Sharma

In International Conference on Machine Learning (ICML), 2025

DOI Bib PDF

@inproceedings{vashishthateaching,
  title = {Teaching Transformers Causal Reasoning through Axiomatic Training},
  author = {Vashishtha, Aniket and Kumar, Abhinav and Pandey, Atharva and Gowtham Reddy, Abbavaram and Ahuja, Kabir and Balasubramanian, Vineeth N and Sharma, Amit},
  booktitle = {International Conference on Machine Learning (ICML)},
  year = {2025},
  publisher = {PMLR},
  doi = {10.48550/arXiv.2407.07612},
}

Causality

Causal Order: The Key to Leveraging Imperfect Experts in Causal Inference

Aniket Vashishtha, Abbavaram Gowtham Reddy, Abhinav Kumar, Saketh Bachu, Vineeth N Balasubramanian, Amit Sharma

In The Thirteenth International Conference on Learning Representations, 2025

DOI Bib PDF

@inproceedings{vashishthacausal,
  title = {Causal Order: The Key to Leveraging Imperfect Experts in Causal Inference},
  author = {Vashishtha, Aniket and Reddy, Abbavaram Gowtham and Kumar, Abhinav and Bachu, Saketh and Balasubramanian, Vineeth N and Sharma, Amit},
  booktitle = {The Thirteenth International Conference on Learning Representations},
  year = {2025},
  doi = {9juyeCqL0u},
}

Testing

Credal Two-sample Tests of Epistemic Ignorance

Siu Lun Chau, Antonin Schrab, Arthur Gretton, Dino Sejdinovic, Krikamol Muandet

In Proceedings of the 18th International Conference on Artificial Intelligence and Statistics (AISTATS), 2025

DOI Bib PDF Poster

@inproceedings{credal2025,
  title = {Credal Two-sample Tests of Epistemic Ignorance},
  author = {Chau, Siu Lun and Schrab, Antonin and Gretton, Arthur and Sejdinovic, Dino and Muandet, Krikamol},
  booktitle = {Proceedings of the 18th International Conference on Artificial Intelligence and Statistics (AISTATS)},
  year = {2025},
  publisher = {PMLR},
  doi = {10.48550/arXiv.2410.12921},
}

Bayesian Opt.

Bayesian Optimization for Building Social-Influence-Free Consensus

Masaki Adachi, Siu Lun Chau, Wenjie Xu, Anurag Singh, Michael A Osborne, Krikamol Muandet

preprint, 2025

DOI Bib PDF

@article{adachi2025bayesian,
  title = {Bayesian Optimization for Building Social-Influence-Free Consensus},
  author = {Adachi, Masaki and Chau, Siu Lun and Xu, Wenjie and Singh, Anurag and Osborne, Michael A and Muandet, Krikamol},
  journal = {preprint},
  year = {2025},
  doi = {10.48550/arXiv.2502.07166},
}

Bayesian Opt.

Highly Parallel Optimisation of Nickel-Catalysed Suzuki Reactions through Automation and Machine Intelligence

Joshua W Sin, Siu Lun Chau, Ryan P Burwood, Kurt Püntener, Raphael Bigler, Philippe Schwaller

Nature Communication, 2025

DOI Bib PDF

@article{sin2024highly,
  title = {Highly Parallel Optimisation of Nickel-Catalysed Suzuki Reactions through Automation and Machine Intelligence},
  author = {Sin, Joshua W and Chau, Siu Lun and Burwood, Ryan P and P{\"u}ntener, Kurt and Bigler, Raphael and Schwaller, Philippe},
  journal = {Nature Communication},
  year = {2025},
  doi = {10.26434/chemrxiv-2024-m12s4}
}

Prediction

Sufficient invariant learning for distribution shift

Taero Kim, Subeen Park, Sungjun Lim, Yonghan Jung, Krikamol Muandet, Kyungwoo Song

Computer Vision and Pattern Recognition (CVPR), 2025

Bib PDF

@article{kim2022sufficient,
  title = {Sufficient invariant learning for distribution shift},
  author = {Kim, Taero and Park, Subeen and Lim, Sungjun and Jung, Yonghan and Muandet, Krikamol and Song, Kyungwoo},
  journal = {Computer Vision and Pattern Recognition (CVPR)},
  year = {2025},
}

2024

Causality

Causal strategic learning with competitive selection

Kiet QH Vo, Muneeb Aadil, Siu Lun Chau, Krikamol Muandet

In Proceedings of the AAAI Conference on Artificial Intelligence, 2024

Oral Bib PDF Code Poster

Oral Presentation at AAAI 2024

@inproceedings{vo2024causal,
  title = {Causal strategic learning with competitive selection},
  author = {Vo, Kiet QH and Aadil, Muneeb and Chau, Siu Lun and Muandet, Krikamol},
  booktitle = {Proceedings of the AAAI Conference on Artificial Intelligence},
  volume = {38},
  number = {14},
  pages = {15411--15419},
  year = {2024},
}

Prediction

Domain Generalisation via Imprecise Learning

Anurag Singh, Siu Lun Chau, Shahine Bouabid, Krikamol Muandet

In International Conference on Machine Learning, 2024

Spotlight Bib PDF Code Poster

Spotlight Presentation at ICML 2024

@inproceedings{singh2024domain,
  title = {Domain Generalisation via Imprecise Learning},
  author = {Singh, Anurag and Chau, Siu Lun and Bouabid, Shahine and Muandet, Krikamol},
  booktitle = {International Conference on Machine Learning},
  pages = {45544--45570},
  year = {2024},
  organization = {PMLR},
}

Prediction

Robust Feature Inference: A Test-time Defense Strategy using Spectral Projections

Anurag Singh*, Mahalakshmi Sabanayagam*, Krikamol Muandet, Debarghya Ghoshdastidar

Transactions on Machine Learning Research, 2024

Bib PDF

@article{singhrobust,
  title = {Robust Feature Inference: A Test-time Defense Strategy using Spectral Projections},
  author = {Singh, Anurag and Sabanayagam, Mahalakshmi and Muandet, Krikamol and Ghoshdastidar, Debarghya},
  journal = {Transactions on Machine Learning Research},
  year = {2024},
}

Causality

Learning Counterfactually Invariant Predictors

Francesco Quinzan, Cecilia Casolo, Krikamol Muandet, Yucen Luo, Niki Kilbertus

Transactions on Machine Learning Research, 2024

Bib PDF

@article{quinzanlearning,
  title = {Learning Counterfactually Invariant Predictors},
  author = {Quinzan, Francesco and Casolo, Cecilia and Muandet, Krikamol and Luo, Yucen and Kilbertus, Niki},
  journal = {Transactions on Machine Learning Research},
  year = {2024},
}

Bayesian Opt.

Looping in the Human: Collaborative and Explainable Bayesian Optimization

Masaki Adachi, Brady Planden, David Howey, Michael A Osborne, Sebastian Orbell, Natalia Ares, Krikamol Muandet, Siu Lun Chau

In International Conference on Artificial Intelligence and Statistics, 2024

DOI Bib PDF Poster

@inproceedings{adachi2024looping,
  title = {Looping in the Human: Collaborative and Explainable Bayesian Optimization},
  author = {Adachi, Masaki and Planden, Brady and Howey, David and Osborne, Michael A and Orbell, Sebastian and Ares, Natalia and Muandet, Krikamol and Chau, Siu Lun},
  booktitle = {International Conference on Artificial Intelligence and Statistics},
  pages = {505--513},
  year = {2024},
  organization = {PMLR},
  doi = {10.48550/arXiv.2310.17273}
}

2023

Causality

Instrumental variable regression via kernel maximum moment loss

Rui Zhang, Masaaki Imaizumi, Bernhard Schölkopf, Krikamol Muandet

Journal of Causal Inference, 2023

DOI Bib PDF Code

@article{zhang2023instrumental,
  title = {Instrumental variable regression via kernel maximum moment loss},
  author = {Zhang, Rui and Imaizumi, Masaaki and Sch{\"o}lkopf, Bernhard and Muandet, Krikamol},
  journal = {Journal of Causal Inference},
  volume = {11},
  number = {1},
  pages = {20220073},
  year = {2023},
  publisher = {De Gruyter},
  doi = {10.1515/jci-2022-0073},
}

Learning

Towards empirical process theory for vector-valued functions: Metric entropy of smooth function classes

Junhyung Park, Krikamol Muandet

In International Conference on Algorithmic Learning Theory, 2023

DOI Bib PDF

@inproceedings{park2023towards,
  title = {Towards empirical process theory for vector-valued functions: Metric entropy of smooth function classes},
  author = {Park, Junhyung and Muandet, Krikamol},
  booktitle = {International Conference on Algorithmic Learning Theory},
  pages = {1216--1260},
  year = {2023},
  organization = {PMLR},
  doi = {https://proceedings.mlr.press/v201/park23a.html},
  slide = {assets/slides/ALT2023-Slides.pdf}
}

Prediction

Gated Domain Units for Multi-source Domain Generalization

Simon Föll*, Alina Dubatovka*, Eugen Ernst†, Siu Lun Chau†, Martin Maritsch, Patrik Okanovic, Gudrun Thaeter, Joachim M Buhmann, Felix Wortmann, Krikamol Muandet

Transactions on Machine Learning Research, 2023

Bib PDF Code

@article{follgated,
  title = {Gated Domain Units for Multi-source Domain Generalization},
  author = {F{\"o}ll, Simon and Dubatovka, Alina and Ernst, Eugen and Chau, Siu Lun and Maritsch, Martin and Okanovic, Patrik and Thaeter, Gudrun and Buhmann, Joachim M and Wortmann, Felix and Muandet, Krikamol},
  journal = {Transactions on Machine Learning Research},
  year = {2023},
}

Causality

A measure-theoretic axiomatisation of causality

Junhyung Park, Simon Buchholz, Bernhard Schölkopf, Krikamol Muandet

Advances in Neural Information Processing Systems, 2023

Oral DOI Bib PDF Poster

Oral Presentation at NeurIPS 2023

@article{park2023measure,
  title = {A measure-theoretic axiomatisation of causality},
  author = {Park, Junhyung and Buchholz, Simon and Sch{\"o}lkopf, Bernhard and Muandet, Krikamol},
  journal = {Advances in Neural Information Processing Systems},
  volume = {36},
  pages = {28510--28540},
  year = {2023},
  doi = {10.48550/arXiv.2305.17139},
}

Explainability

Explaining the Uncertain: Stochastic Shapley Values for Gaussian Process Models

Siu Lun Chau, Krikamol Muandet*, Dino Sejdinovic*

Advances in Neural Information Processing Systems, 2023

Spotlight Bib PDF Code Poster

Spotlight Presentation at NeurIPS 2023

@article{chau2023explaining,
  title = {Explaining the Uncertain: Stochastic Shapley Values for Gaussian Process Models},
  author = {Chau, Siu Lun and Muandet, Krikamol and Sejdinovic, Dino},
  journal = {Advances in Neural Information Processing Systems},
  volume = {36},
  pages = {50769--50795},
  year = {2023},
}

Causality

On the Relationship Between Explanation and Prediction: A Causal View

Amir-Hossein Karimi, Krikamol Muandet, Simon Kornblith, Bernhard Schölkopf, Been Kim

In International Conference on Machine Learning, 2023

Bib PDF Poster

@inproceedings{karimi2023relationship,
  title = {On the Relationship Between Explanation and Prediction: A Causal View},
  author = {Karimi, Amir-Hossein and Muandet, Krikamol and Kornblith, Simon and Sch{\"o}lkopf, Bernhard and Kim, Been},
  booktitle = {International Conference on Machine Learning},
  pages = {15861--15883},
  year = {2023},
  organization = {PMLR},
}

2022

Learning

Impossibility of collective intelligence

Krikamol Muandet

arXiv preprint arXiv:2206.02786, 2022

Bib PDF

@article{muandet2022impossibility,
  title = {Impossibility of collective intelligence},
  author = {Muandet, Krikamol},
  journal = {arXiv preprint arXiv:2206.02786},
  year = {2022},
}