← Search

Tom Diethe

13 accepted papers

2026

Causal-Adapter: Taming Text-to-Image Diffusion for Faithful Counterfactual Generation

ICML 2026poster

We present Causal-Adapter, a modular framework that adapts frozen text-to-image diffusion backbones for counterfactual image generation. Our method enables causal interventions on target attributes while preserving all other aspects of the image, including the core identity. In contrast to prior app…

Cited by 0SourceScholar
2025

Balancing Act: Diversity and Consistency in Large Language Model Ensembles

ICLR 2025poster

Ensembling strategies for Large Language Models (LLMs) have demonstrated significant potential in improving performance across various tasks by combining the strengths of individual models. However, identifying the most effective ensembling method remains an open challenge, as neither maximizing out…

Cited by 0SourcePDFScholar
2025

DeCoRe: Decoding by Contrasting Retrieval Heads to Mitigate Hallucinations

EMNLP 2025

Large Language Models (LLMs) often hallucinate, producing unfaithful or factually incorrect outputs by misrepresenting the provided context or incorrectly recalling internal knowledge. Recent studies have identified specific attention heads within the Transformer architecture, known as retrieval hea

2025

Diffusion Instruction Tuning

ICML 2025poster

We introduce *Lavender*, a simple supervised fine-tuning (SFT) method that boosts the performance of advanced vision-language models (VLMs) by leveraging state-of-the-art image generation models such as Stable Diffusion. Specifically, Lavender aligns the text-vision attention in the VLM transformer…

2025

Return of the Latent Space COWBOYS: Re-thinking the use of VAEs for Bayesian Optimisation of Structured Spaces

ICML 2025spotlight

Bayesian optimisation in the latent space of a VAE is a powerful framework for optimisation tasks over complex structured domains, such as the space of valid molecules. However, existing approaches tightly couple the surrogate and generative models, which can lead to suboptimal performance when the…

Cited by 0SourcePDFScholar
2025

Segment Anyword: Mask Prompt Inversion for Open-Set Grounded Segmentation

ICML 2025poster

Open-set image segmentation poses a significant challenge because existing methods often demand extensive training or fine-tuning and generally struggle to segment unified objects consistently across diverse text reference expressions. Motivated by this, we propose Segment Anyword, a novel training-…

Cited by 0SourcePDFScholar
2024

An Image is Worth Multiple Words: Discovering Object Level Concepts using Multi-Concept Prompt Learning

ICML 2024poster

Textural Inversion, a prompt learning method, learns a singular text embedding for a new "word" to represent image style and appearance, allowing it to be integrated into natural language sentences to generate novel synthesised images. However, identifying multiple unknown object-level concepts with…

2024

Improving Antibody Humanness Prediction using Patent Data

ICML 2024poster

We investigate the potential of patent data for improving the antibody humanness prediction using a multi-stage, multi-loss training process. Humanness serves as a proxy for the immunogenic response to antibody therapeutics, one of the major causes of attrition in drug discovery and a challenging ob…

2024

Measures of diversity and space-filling designs for categorical data

ICML 2024poster

Selecting a small subset of items that represent the diversity of a larger population lies at the heart of many data analysis and machine learning applications. However, when it comes to items described by discrete features, the lack of natural ordering and the combinatorial nature of the search spa…

Cited by 0SourcePDFScholar
2024

Tackling Structural Hallucination in Image Translation with Local Diffusion

ECCV 2024oral

"Recent developments in diffusion models have advanced conditioned image generation, yet they struggle with reconstructing out-of-distribution (OOD) images, such as unseen tumors in medical images, causing “image hallucination” and risking misdiagnosis. We hypothesize such hallucinations result from…

2019

$β^3$-IRT: A New Item Response Model and its Applications

AISTATS 2019poster

Item Response Theory (IRT) aims to assess latent abilities of respondents based on the correctness of their answers in aptitude test items with different difficulty levels. In this paper, we propose the $\beta^3$-IRT model, which models continuous responses and can generate a much enriched family o…