← Search

Kareem Ahmed

10 accepted papers

2024

Probabilistically Rewired Message-Passing Neural Networks

ICLR 2024poster

Message-passing graph neural networks (MPNNs) emerged as powerful tools for processing graph-structured input. However, they operate on a fixed input graph structure, ignoring potential noise and missing information. Furthermore, their local aggregation mechanism can lead to problems such as over-sq…

2024

Scaling Tractable Probabilistic Circuits: A Systems Perspective

ICML 2024poster

Probabilistic Circuits (PCs) are a general framework for tractable deep generative models, which support exact and efficient probabilistic inference on their learned distributions. Recent modeling and training advancements have enabled their application to complex real-world tasks. However, the time…

2024

Where is the signal in tokenization space?

EMNLP 2024main

Large Language Models (LLMs) are typically shipped with tokenizers that *deterministically* encode text into so-called *canonical* token sequences, to which the LLMs assign probability values.One common assumption is that the probability of a piece of text is the probability of its canonical token s…

Cited by 3SourcePDFScholar
2023

A Pseudo-Semantic Loss for Autoregressive Models with Logical Constraints

NeurIPS 2023poster

Neuro-symbolic AI bridges the gap between purely symbolic and neural approaches to learning. This often requires maximizing the likelihood of a symbolic constraint w.r.t the neural network's output distribution. Such output distributions are typically assumed to be fully-factorized. This limits the…

Cited by 15SourcePDFScholar
2023

A Unified Approach to Count-Based Weakly Supervised Learning

NeurIPS 2023poster

High-quality labels are often very scarce, whereas unlabeled data with inferred weak labels occurs more naturally. In many cases, these weak labels dictate the frequency of each respective class over a set of instances. In this paper, we develop a unified approach to learning from such weakly-labele…

2023

SIMPLE: A Gradient Estimator for k-Subset Sampling

ICLR 2023poster

$k$-subset sampling is ubiquitous in machine learning, enabling regularization and interpretability through sparsity. The challenge lies in rendering $k$-subset sampling amenable to end-to-end learning. This has typically involved relaxing the reparameterized samples to allow for backpropagation, bu…

2022

Semantic Probabilistic Layers for Neuro-Symbolic Learning

NeurIPS 2022accept

We design a predictive layer for structured-output prediction (SOP) that can be plugged into any neural network guaranteeing its predictions are consistent with a set of predefined symbolic constraints. Our Semantic Probabilistic Layer (SPL) can model intricate correlations, and hard constraints, ov…