← Search

Sohir Maskey

6 accepted papers

2026

1-Bit Wonder: Improving QAT Performance in the Low-Bit Regime through K-Means Quantization

ICML 2026poster

Quantization-aware training (QAT) is an effective method to drastically reduce the memory footprint of LLMs while keeping performance degradation at an acceptable level. However, the optimal choice of quantization format and bit-width presents a challenge in practice. The full design space of quanti…

Cited by 0SourceScholar
2026

Graph Representational Learning: When Does More Expressivity Hurt Generalization?

ICLR 2026poster

Graph Neural Networks (GNNs) are powerful tools for learning on structured data, yet the relationship between their expressivity and predictive performance remains unclear. We introduce a family of pseudometrics that capture different degrees of structural similarity between graphs and relate these…

Cited by 0SourcecodeScholar
2026

The Price of Robustness: Stable Classifiers Need Overparameterization

ICLR 2026poster

The relationship between overparameterization, stability, and generalization remains incompletely understood in the setting of discontinuous classifiers. We address this gap by establishing a generalization bound for finite function classes that improves inversely with _class stability_, defined…

Cited by 0SourceScholar
2024

Weisfeiler and Leman Go Loopy: A New Hierarchy for Graph Representational Learning

NeurIPS 2024oral

We introduce $r$-loopy Weisfeiler-Leman ($r$-$\ell$WL), a novel hierarchy of graph isomorphism tests and a corresponding GNN framework, $r$-$\ell$MPNN, that can count cycles up to length $r{+}2$. Most notably, we show that $r$-$\ell$WL can count homomorphisms of cactus graphs. This extends 1-WL, whi…

2023

A Fractional Graph Laplacian Approach to Oversmoothing

NeurIPS 2023poster

Graph neural networks (GNNs) have shown state-of-the-art performances in various applications. However, GNNs often struggle to capture long-range dependencies in graphs due to oversmoothing. In this paper, we generalize the concept of oversmoothing from undirected to directed graphs. To this aim, we…

2022

Generalization Analysis of Message Passing Neural Networks on Large Random Graphs

NeurIPS 2022accept

Message passing neural networks (MPNN) have seen a steep rise in popularity since their introduction as generalizations of convolutional neural networks to graph-structured data, and are now considered state-of-the-art tools for solving a large variety of graph-focused problems. We study the general…

Cited by 74SourcePDFScholar