3 papers
cs.LG2026
WildCat: Near-Linear Attention in Theory and Practice
Tobias Schröder, Lester Mackey
We introduce WildCat, a high-accuracy, low-cost approach to compressing the attention mechanism in neural networks. While attention is a staple of modern network architectures, it…
stat.ML2025
Deep Optimal Sensor Placement for Black Box Stochastic Simulations
Paula Cordero-Encinar, Tobias Schröder, Peter Yatsyshin +1
Selecting cost-effective optimal sensor configurations for subsequent inference of parameters in black-box stochastic systems faces significant computational barriers. We propose a…
stat.ML2024
Energy-Based Modelling for Discrete and Mixed Data via Heat Equations on Structured Spaces
Tobias Schröder, Zijing Ou, Yingzhen Li +1
Energy-based models (EBMs) offer a flexible framework for probabilistic modelling across various data domains. However, training EBMs on data in discrete or mixed state spaces pose…