← Search

Shubhendu Trivedi

15 accepted papers

2026

On Universality of Deep Equivariant Networks

ICLR 2026poster

Universality results for equivariant neural networks remain rare. Those that do exist typically hold only in restrictive settings: either they rely on regular or higher-order tensor representations, leading to impractically high-dimensional hidden spaces, or they target specialized architectures, of…

Cited by 0SourceScholar
2026

Recurrent Equivariant Constraint Modulation: Learning Per-Layer Symmetry Relaxation from Data

ICML 2026spotlight

Equivariant neural networks exploit underlying task symmetries to improve generalization, but strict equivariance constraints can induce more complex optimization dynamics that can hinder learning. Prior work addresses these limitations by relaxing strict equivariance during training, but typically …

Cited by 0SourceScholar
2025

Symmetry-Based Structured Matrices for Efficient Approximately Equivariant Networks

AISTATS 2025oral

There has been much recent interest in designing neural networks (NNs) with relaxed equivariance, which interpolate between exact equivariance and full flexibility for consistent performance gains. In a separate line of work, structured parameter matrices with low displacement rank (LDR)---which per…

Cited by 0SourcecodeScholar
2024

Contextualized Sequence Likelihood: Enhanced Confidence Scores for Natural Language Generation

EMNLP 2024main

The advent of large language models (LLMs) has dramatically advanced the state-of-the-art in numerous natural language generation tasks. For LLMs to be applied reliably, it is essential to have an accurate measure of their confidence. Currently, the most commonly used confidence score function is th…

2024

Improving Equivariant Model Training via Constraint Relaxation

NeurIPS 2024poster

Equivariant neural networks have been widely used in a variety of applications due to their ability to generalize well in tasks where the underlying data symmetries are known. Despite their successes, such networks can be difficult to optimize and require careful hyperparameter tuning to train succe…

2023

Approximation-Generalization Trade-offs under (Approximate) Group Equivariance

NeurIPS 2023poster

The explicit incorporation of task-specific inductive biases through symmetry has emerged as a general design precept in the development of high-performance machine learning models. For example, group equivariant neural networks have demonstrated impressive performance across various domains and app…

Cited by 22SourcePDFScholar
2023

Fast Online Value-Maximizing Prediction Sets with Conformal Cost Control

ICML 2023poster

Many real-world multi-label prediction problems involve set-valued predictions that must satisfy specific requirements dictated by downstream usage. We focus on a typical scenario where such requirements, separately encoding *value* and *cost*, compete with each other. For instance, a hospital might…

2023

Taking a Step Back with KCal: Multi-Class Kernel-Based Calibration for Deep Neural Networks

ICLR 2023poster

Deep neural network (DNN) classifiers are often overconfident, producing miscalibrated class probabilities. In high-risk applications like healthcare, practitioners require fully calibrated probability predictions for decision-making. That is, conditioned on the prediction vector, every class’ proba…

Cited by 7SourcePDFScholar
2022

Capacity of Group-invariant Linear Readouts from Equivariant Representations: How Many Objects can be Linearly Classified Under All Possible Views?

ICLR 2022poster

Equivariance has emerged as a desirable property of representations of objects subject to identity-preserving transformations that constitute a group, such as translations and rotations. However, the expressivity of a representation constrained by group equivariance is still not fully understood. We…

2021

Locally Valid and Discriminative Prediction Intervals for Deep Learning Models

NeurIPS 2021poster

Crucial for building trust in deep learning models for critical real-world applications is efficient and theoretically sound uncertainty quantification, a task that continues to be challenging. Useful uncertainty information is expected to have two key properties: It should be valid (guaranteeing co…

2018

Clebsch–Gordan Nets: a Fully Fourier Space Spherical Convolutional Neural Network

NeurIPS 2018poster

Recent work by Cohen et al. has achieved state-of-the-art results for learning spherical images in a rotation invariant way by using ideas from group representation theory and noncommutative harmonic analysis. In this paper we propose a generalization of this work that generally exhibits improved pe…

Cited by 327SourcePDFScholar
2018

Covariant Compositional Networks For Learning Graphs

ICLR 2018workshop

Most existing neural networks for learning graphs deal with the issue of permutation invariance by conceiving of the network as a message passing scheme, where each node sums the feature vectors coming from its neighbors. We argue that this imposes a limitation on their representation power, and ins…

Cited by 157SourcecodeScholar
2018

On the Generalization of Equivariance and Convolution in Neural Networks to the Action of Compact Groups

ICML 2018oral

Convolutional neural networks have been extremely successful in the image recognition domain because they ensure equivariance with respect to translations. There have been many recent attempts to generalize this framework to other domains, including graphs and data lying on manifolds. In this paper…

Cited by 602SourcePDFScholar