← Search

Amauri H Souza

15 accepted papers

2026

Differentiable Lifting for Topological Neural Networks

ICLR 2026poster

Topological neural networks (TNNs) enable leveraging high-order structures on graphs (e.g., cycles and cliques) to boost the expressive power of message-passing neural networks. In turn, however, these structures are typically identified a priori through an unsupervised graph lifting operation. Notw…

Cited by 0SourcecodeScholar
2025

Generalization and Distributed Learning of GFlowNets

ICLR 2025poster

Conventional wisdom attributes the success of Generative Flow Networks (GFlowNets) to their ability to exploit the compositional structure of the sample space for learning generalizable flow functions (Bengio et al., 2021). Despite the abundance of empirical evidence, formalizing this belief with ve…

Cited by 0SourcePDFScholar
2025

When do GFlowNets learn the right distribution?

ICLR 2025spotlight

Generative Flow Networks (GFlowNets) are an emerging class of sampling methods for distributions over discrete and compositional objects, e.g., graphs. In spite of their remarkable success in problems such as drug discovery and phylogenetic inference, the question of when and whether GFlowNets learn…

Cited by 1SourcePDFScholar
2024

Compositional PAC-Bayes: Generalization of GNNs with persistence and beyond

NeurIPS 2024poster

Heterogeneity, e.g., due to different types of layers or multiple sub-models, poses key challenges in analyzing the generalization behavior of several modern architectures. For instance, descriptors based on Persistent Homology (PH) are being increasingly integrated into Graph Neural Networks (GNNs)…

2023

Learning Robust Statistics for Simulation-based Inference under Model Misspecification

NeurIPS 2023poster

Simulation-based inference (SBI) methods such as approximate Bayesian computation (ABC), synthetic likelihood, and neural posterior estimation (NPE) rely on simulating statistics to infer parameters of intractable likelihood models. However, such methods are known to yield untrustworthy and mislead…

2022

Provably expressive temporal graph networks

NeurIPS 2022accept

Temporal graph networks (TGNs) have gained prominence as models for embedding dynamic interactions, but little is known about their theoretical underpinnings. We establish fundamental results about the representational power and limits of the two main categories of TGNs: those that aggregate tempor…