← Search

Pramod Viswanath

27 accepted papers

2026

AutoCode: LLMs as Problem Setters for Competitive Programming

ICLR 2026poster

Writing competitive programming problems is exacting. Authors must: set constraints, input distributions, and edge cases that rule out shortcuts; target specific algorithms (e.g., max-flow, dynamic programming, data structures); and calibrate complexity beyond the reach of most competitors. We argue…

Cited by 0SourceScholar
2026

FrontierCS: Evolving Challenges for Evolving Intelligence

ICML 2026poster

We introduce FrontierCS, a benchmark of 240 open-ended problems across diverse areas of computer science, designed and reviewed by experts, including CS PhDs and top-tier competitive programming participants and problem setters. Unlike existing benchmarks that focus on tasks with known optimal solut…

Cited by 0SourceScholar
2026

MEMO: Memory-Augmented Model Context Optimization for Robust Multi-Turn Multi-Agent LLM Games

ICML 2026poster

Multi-turn, multi-agent LLM game evaluations often exhibit substantial run-to-run variance. In long-horizon interactions, small early deviations compound across turns and are amplified by multi-agent coupling, biasing win rate estimates and destabilizing comparative rankings across repeated tourname…

Cited by 0SourcecodeScholar
2026

TritonGym: A Benchmark for Agentic LLM Workflows in Triton GPU Code Generation

ICML 2026poster

Large language models (LLMs) can already draft plausible Triton kernels, yet most existing evaluations still focus on single-shot generation and underplay tool use and feedback. We introduce *TritonGym*, a benchmark and orchestration framework for evaluating agentic workflows in GPU code generation.…

Cited by 0SourceScholar
2025

LiveCodeBench Pro: How Do Olympiad Medalists Judge LLMs in Competitive Programming?

NeurIPS 2025poster

Recent reports claim that large language models (LLMs) now outperform elite humans in competitive programming. Drawing on knowledge from a group of medalists in international algorithmic contests, we revisit this claim, examining how LLMs differ from human experts and where limitations still remain.…

Cited by 0SourceScholar
2025

Scalable Fingerprinting of Large Language Models

NeurIPS 2025spotlight

Model fingerprinting has emerged as a powerful tool for model owners to identify their shared model given API access. In order to lower false discovery rate, fight fingerprint leakage, and defend against coalitions of model users attempting to bypass detection, we argue that scaling up the number of…

Cited by 0SourceScholar
2024

DeepPolar: Inventing Nonlinear Large-Kernel Polar Codes via Deep Learning

ICML 2024poster

Progress in designing channel codes has been driven by human ingenuity and, fittingly, has been sporadic. Polar codes, developed on the foundation of Arikan’s polarization kernel, represent the latest breakthrough in coding theory and have emerged as the state-of-the-art error-correction code for sh…

2023

CRISP: Curriculum based Sequential neural decoders for Polar code family

ICML 2023poster

Polar codes are widely used state-of-the-art codes for reliable communication that have recently been included in the $5^{\text{th}}$ generation wireless standards ($5$G). However, there still remains room for design of polar decoders that are both efficient and reliable in the short blocklength reg…

2021

KO codes: inventing nonlinear encoding and decoding for reliable wireless communication via deep-learning

ICML 2021spotlight

Landmark codes underpin reliable physical layer communication, e.g., Reed-Muller, BCH, Convolution, Turbo, LDPC, and Polar codes: each is a linear code and represents a mathematical breakthrough. The impact on humanity is huge: each of these codes has been used in global wireless communication stand…

2019

Breaking the gridlock in Mixture-of-Experts: Consistent and Efficient Algorithms

ICML 2019oral

Mixture-of-Experts (MoE) is a widely popular model for ensemble learning and is a basic building block of highly successful modern neural networks as well as a component in Gated Recurrent Units (GRU) and Attention networks. However, present algorithms for learning MoE, including the EM algorithm an…

Cited by 33SourcePDFScholar
2019

Learning One-hidden-layer Neural Networks under General Input Distributions

AISTATS 2019poster

Significant advances have been made recently on training neural networks, where the main challenge is in solving an optimization problem with abundant critical points. However, existing approaches to address this issue crucially rely on a restrictive assumption: the training data is drawn from a Gau…

Cited by 35SourcePDFScholar
2019

Turbo Autoencoder: Deep learning based channel codes for point-to-point communication channels

NeurIPS 2019poster

Designing codes that combat the noise in a communication medium has remained a significant area of research in information theory as well as wireless communications. Asymptotically optimal channel codes have been developed by mathematicians for communicating under canonical models after over 60 year…

2018

Communication Algorithms via Deep Learning

ICLR 2018poster

Coding theory is a central discipline underpinning wireline and wireless modems that are the workhorses of the information age. Progress in coding theory is largely driven by individual human ingenuity with sporadic breakthroughs over the past century. In this paper we study whether it is possible t…

2018

Deepcode: Feedback Codes via Deep Learning

NeurIPS 2018poster

The design of codes for communicating reliably over a statistically well defined channel is an important endeavor involving deep mathematical research and wide- ranging practical applications. In this work, we present the first family of codes obtained via deep learning, which significantly beats st…

2018

Estimators for Multivariate Information Measures in General Probability Spaces

NeurIPS 2018poster

Information theoretic quantities play an important role in various settings in machine learning, including causality testing, structure inference in graphical models, time-series problems, feature selection as well as in providing privacy guarantees. A key quantity of interest is the mutual informat…

Cited by 19SourcePDFScholar
2017

Discovering Potential Correlations via Hypercontractivity

NeurIPS 2017poster

Discovering a correlation from one variable to another variable is of fundamental scientific and practical interest. While existing correlation measures are suitable for discovering average correlation, they fail to discover hidden or potential correlations. To bridge this gap, (i) we postulate a se…

2017

Estimating Mutual Information for Discrete-Continuous Mixtures

NeurIPS 2017spotlight

Estimation of mutual information from observed samples is a basic primitive in machine learning, useful in several learning tasks including correlation mining, information bottleneck, Chow-Liu tree, and conditional independence testing in (causal) graphical models. While mutual information is a quan…

Cited by 213SourcePDFScholar
2017

Geometry of Polysemy

ICLR 2017poster

Vector representations of words have heralded a transformational approach to classical problems in NLP; the most popular example is word2vec. However, a single vector does not suffice to model the polysemous nature of many (frequent) words, i.e., words with multiple meanings. In this paper, we pr…

Cited by 33SourceScholar
2016

Breaking the Bandwidth Barrier: Geometrical Adaptive Entropy Estimation

NeurIPS 2016poster

Estimators of information theoretic measures such as entropy and mutual information from samples are a basic workhorse for many downstream applications in modern data science. State of the art approaches have been either geometric (nearest neighbor (NN) based) or kernel based (with bandwidth chosen…

Cited by 42SourcePDFScholar
2016

Conditional Dependence via Shannon Capacity: Axioms, Estimators and Applications

ICML 2016poster

We consider axiomatically the problem of estimating the strength of a conditional dependence relationship P_Y|X from a random variables X to a random variable Y. This has applications in determining the strength of a known causal relationship, where the strength depends only on the conditional distr…

Cited by 11SourcePDFScholar
2016

Metadata-conscious anonymous messaging

ICML 2016poster

Anonymous messaging platforms like Whisper and Yik Yak allow users to spread messages over a network (e.g., a social network) without revealing message authorship to other users. The spread of messages on these platforms can be modeled by a diffusion process over a graph. Recent advances in network…

Cited by 12SourcePDFScholar