← Search

Tom Wollschläger

10 accepted papers

2025

Lift Your Molecules: Molecular Graph Generation in Latent Euclidean Space

ICLR 2025poster

We introduce a new framework for 2D molecular graph generation using 3D molecule generative models. Our Synthetic Coordinate Embedding (SyCo) framework maps 2D molecular graphs to 3D Euclidean point clouds via synthetic coordinates and learns the inverse map using an E($n$)-Equivariant Graph Neural…

Cited by 1SourcePDFScholar
2025

REINFORCE Adversarial Attacks on Large Language Models: An Adaptive, Distributional, and Semantic Objective

ICML 2025poster

To circumvent the alignment of large language models (LLMs), current optimization-based adversarial attacks usually craft adversarial prompts by maximizing the likelihood of a so-called affirmative response. An affirmative response is a manually designed start of a harmful answer to an inappropriate…

2025

The Geometry of Refusal in Large Language Models: Concept Cones and Representational Independence

ICML 2025poster

The safety alignment of large language models (LLMs) can be circumvented through adversarially crafted inputs, yet the mechanisms by which these attacks bypass safety barriers remain poorly understood. Prior work suggests that a *single* refusal direction in the model's activation space determines w…

Cited by 0SourcePDFScholar
2025

Uncertainty Estimation for Heterophilic Graphs Through the Lens of Information Theory

ICML 2025poster

While uncertainty estimation for graphs recently gained traction, most methods rely on homophily and deteriorate in heterophilic settings. We address this by analyzing message passing neural networks from an information-theoretic perspective and developing a suitable analog to data processing in…

Cited by 0SourcePDFScholar
2025

What Expressivity Theory Misses: Message Passing Complexity for GNNs

NeurIPS 2025spotlight

Expressivity theory, characterizing which graphs a GNN can distinguish, has become the predominant framework for analyzing GNNs, with new models striving for higher expressivity. However, we argue that this focus is misguided: First, higher expressivity is not necessary for most real-world tasks as…

Cited by 0SourceScholar
2024

Energy-based Epistemic Uncertainty for Graph Neural Networks

NeurIPS 2024spotlight

In domains with interdependent data, such as graphs, quantifying the epistemic uncertainty of a Graph Neural Network (GNN) is challenging as uncertainty can arise at different structural scales. Existing techniques neglect this issue or only distinguish between structure-aware and structure-agnostic…

Cited by 1SourcePDFScholar
2024

Expressivity and Generalization: Fragment-Biases for Molecular GNNs

ICML 2024oral

Although recent advances in higher-order Graph Neural Networks (GNNs) improve the theoretical expressiveness and molecular property predictive performance, they often fall short of the empirical performance of models that explicitly use fragment information as inductive bias. However, for these appr…

Cited by 5SourcePDFScholar
2024

Uncertainty for Active Learning on Graphs

ICML 2024poster

Uncertainty Sampling is an Active Learning strategy that aims to improve the data efficiency of machine learning models by iteratively acquiring labels of data points with the highest uncertainty. While it has proven effective for independent data its applicability to graphs remains under-explored.…

Cited by 10SourcePDFScholar
2023

Localized Randomized Smoothing for Collective Robustness Certification

ICLR 2023top-25%

Models for image segmentation, node classification and many other tasks map a single input to multiple labels. By perturbing this single shared input (e.g. the image) an adversary can manipulate several predictions (e.g. misclassify several pixels). Collective robustness certification is the task of…

Cited by 11SourcePDFScholar
2023

Uncertainty Estimation for Molecules: Desiderata and Methods

ICML 2023poster

Graph Neural Networks (GNNs) are promising surrogates for quantum mechanical calculations as they establish unprecedented low errors on collections of molecular dynamics (MD) trajectories. Thanks to their fast inference times they promise to accelerate computational chemistry applications. Unfortuna…

Cited by 13SourcePDFScholar