← Search

Mislav Balunovic

18 accepted papers

2026

The Open Proof Corpus: A Large-Scale Study of LLM-Generated Mathematical Proofs

ICLR 2026poster

In recent months, large language models (LLMs) have made significant progress in mathematical proof generation, but further advancement is hindered by the lack of a large-scale, high-quality dataset of human-evaluated proofs. While expensive to create, such a dataset is essential for driving improve…

Cited by 0SourcecodeScholar
2025

MathArena: Evaluating LLMs on Uncontaminated Math Competitions

NeurIPS 2025poster

The rapid advancement of reasoning capabilities in large language models (LLMs) has led to notable improvements on mathematical benchmarks. However, many of the most commonly used evaluation datasets (e.g., AIME 2024) are widely available online, making it difficult to disentangle genuine reasoning…

Cited by 0SourcecodeScholar
2025

MathConstruct: Challenging LLM Reasoning with Constructive Proofs

ICML 2025poster

While Large Language Models (LLMs) demonstrate impressive performance in mathematics, existing math benchmarks come with significant limitations. Many focus on problems with fixed ground-truth answers, and are often saturated due to problem simplicity or the viability of guessing or memorization. Cr…

Cited by 1SourcePDFScholar
2024

AgentDojo: A Dynamic Environment to Evaluate Prompt Injection Attacks and Defenses for LLM Agents

NeurIPS 2024poster

AI agents aim to solve complex tasks by combining text-based reasoning with external tool calls. Unfortunately, AI agents are vulnerable to prompt injection attacks where data returned by external tools hijacks the agent to execute malicious tasks. To measure the adversarial robustness of AI agents,…

2024

Beyond Memorization: Violating Privacy via Inference with Large Language Models

ICLR 2024spotlight

Current privacy research on large language models (LLMs) primarily focuses on the issue of extracting memorized training data. At the same time, models’ inference capabilities have increased drastically. This raises the key question of whether current LLMs could violate individuals’ privacy by infer…

2023

FARE: Provably Fair Representation Learning with Practical Certificates

ICML 2023poster

Fair representation learning (FRL) is a popular class of methods aiming to produce fair classifiers via data preprocessing. Recent regulatory directives stress the need for FRL methods that provide practical certificates, i.e., provable upper bounds on the unfairness of any downstream classifier tra…

2023

TabLeak: Tabular Data Leakage in Federated Learning

ICML 2023poster

While federated learning (FL) promises to preserve privacy, recent works in the image and text domains have shown that training updates leak private client data. However, most high-stakes applications of FL (e.g., in healthcare and finance) use tabular data, where the risk of data leakage has not ye…

2022

Bayesian Framework for Gradient Leakage

ICLR 2022poster

Federated learning is an established method for training machine learning models without sharing training data. However, recent work has shown that it cannot guarantee data privacy as shared gradients can still leak sensitive information. To formalize the problem of gradient leakage, we propose a th…

2022

LAMP: Extracting Text from Gradients with Language Model Priors

NeurIPS 2022accept

Recent work shows that sensitive user data can be reconstructed from gradient updates, breaking the key privacy promise of federated learning. While success was demonstrated primarily on image data, these methods do not directly transfer to other domains such as text. In this work, we propose LAMP,…

2021

Certify or Predict: Boosting Certified Robustness with Compositional Architectures

ICLR 2021poster

A core challenge with existing certified defense mechanisms is that while they improve certified robustness, they also tend to drastically decrease natural accuracy, making it difficult to use these methods in practice. In this work, we propose a new architecture which addresses this challenge and e…

Cited by 17SourcePDFScholar
2020

Learning Certified Individually Fair Representations

NeurIPS 2020poster

Fair representation learning provides an effective way of enforcing fairness constraints without compromising utility for downstream users. A desirable family of such fairness constraints, each requiring similar treatment for similar individuals, is known as individual fairness. In this work, we int…

2019

Certifying Geometric Robustness of Neural Networks

NeurIPS 2019poster

The use of neural networks in safety-critical computer vision systems calls for their robustness certification against natural geometric transformations (e.g., rotation, scaling). However, current certification methods target mostly norm-based pixel perturbations and cannot certify robustness agains…

2019

DL2: Training and Querying Neural Networks with Logic

ICML 2019oral

We present DL2, a system for training and querying neural networks with logical constraints. Using DL2, one can declaratively specify domain knowledge constraints to be enforced during training, as well as pose queries on the model to find inputs that satisfy a set of constraints. DL2 works by trans…