Manzil Zaheer

Efficient Distributed Optimization under Heavy-Tailed Noise ICML 2025 Adaptive Retrieval and Scalable Indexing for k-NN Search with Cross-Encoders ICLR 2024 A Fresh Take on Stale Embeddings: Improving Dense Retriever Training with Corrector Networks ICML 2024 Analysis of Plan-based Retrieval for Grounded Text Generation EMNLP 2024 A Statistical Framework for Data-dependent Retrieval-Augmented Models ICML 2024 Can Public Large Language Models Help Private Cross-device Federated Learning? NAACL 2024 ExeDec: Execution Decomposition for Compositional Generalization in Neural Program Synthesis ICLR 2024 Functional Interpolation for Relative Positions improves Long Context Transformers ICLR 2024 USTAD: Unified Single-model Training Achieving Diverse Scores for Information Retrieval ICML 2024 A Statistical Perspective on Retrieval-Based Models ICML 2023 Differentially Private Adaptive Optimization with Delayed Preconditioners ICLR 2023 Efficient k-NN Search with Cross-Encoders using Adaptive Multi-Round CUR Decomposition EMNLP 2023 Improving Dual-Encoder Training through Dynamic Indexes for Negative Mining AISTATS 2023 Large Language Models with Controllable Working Memory ACL 2023 Longtonotes: OntoNotes with Longer Coreference Chains EACL 2023 Machine Reading Comprehension using Case-based Reasoning EMNLP 2023 Multi-Task Off-Policy Learning from Bandit Feedback ICML 2023 ResMem: Learn what you can and memorize the rest NeurIPS 2023 Teacher Guided Training: An Efficient Framework for Knowledge Transfer ICLR 2023 A Context-Integrated Transformer-Based Neural Network for Auction Design ICML 2022 A Fourier Approach to Mixture Learning NeurIPS 2022 Deep Hierarchy in Bandits ICML 2022 Efficient Nearest Neighbor Search for Cross-Encoder Models using Matrix Factorization EMNLP 2022 Hierarchical Bayesian Bandits AISTATS 2022 Knowledge Base Question Answering by Case-based Reasoning over Subgraphs ICML 2022 Learning to Navigate Wikipedia by Taking Random Walks NeurIPS 2022 Private Adaptive Optimization with Side information ICML 2022 Robust Training of Neural Networks Using Scale Invariant Architectures ICML 2022 StreamingQA: A Benchmark for Adaptation to New Knowledge over Time in Question Answering Models ICML 2022 Thompson Sampling with a Mixture Prior AISTATS 2022 Adaptive Federated Optimization ICLR 2021 Anchor & Transform: Learning Sparse Embeddings for Large Vocabularies ICLR 2021 Case-based Reasoning for Natural Language Queries over Knowledge Bases EMNLP 2021 DAG-Structured Clustering by Nearest Neighbors AISTATS 2021 Differentiable Open-Ended Commonsense Reasoning NAACL 2021 Exact and approximate hierarchical clustering using A* UAI 2021 Federated Composite Optimization ICML 2021 Latent Programmer: Discrete Latent Codes for Program Synthesis ICML 2021 Meta-Thompson Sampling ICML 2021 Non-Stationary Off-Policy Optimization AISTATS 2021 No Regrets for Learning the Prior in Bandits NeurIPS 2021 Scaling Within Document Coreference to Long Texts ACL 2021 Sketch based Memory for Neural Networks AISTATS 2021 Unsupervised Abstractive Dialogue Summarization for Tete-a-Tetes AAAI 2021 Big Bird: Transformers for Longer Sequences NeurIPS 2020 Differentiable Meta-Learning of Bandit Policies NeurIPS 2020 Differentiable Reasoning over a Virtual Knowledge Base ICLR 2020 Exchangeable Generative Models with Flow Scans AAAI 2020 Latent Bandits Revisited NeurIPS 2020 PLLay: Efficient Topological Layer based on Persistent Landscapes NeurIPS 2020 Probabilistic Case-based Reasoning for Open-World Knowledge Graph Completion EMNLP 2020 Randomized Exploration in Generalized Linear Bandits AISTATS 2020 Robust large-margin learning in hyperbolic space NeurIPS 2020 Chains-of-Reasoning at TextGraphs 2019 Shared Task: Reasoning over Chains of Facts for Explainable Multi-hop Inference EMNLP 2019 Multi-step Entity-centric Information Retrieval for Multi-Hop Question Answering EMNLP 2019 Multi-step Retriever-Reader Interaction for Scalable Open-domain Question Answering ICLR 2019 Revisiting LSTM Networks for Semi-Supervised Text Classification via Mixed Objective Function AAAI 2019 Towards Gradient Free and Projection Free Stochastic Optimization AISTATS 2019 Adaptive Methods for Nonconvex Optimization NeurIPS 2018 A Generic Approach for Escaping Saddle points AISTATS 2018 Compressed Video Action Recognition CVPR 2018 Go for a Walk and Arrive at the Answer: Reasoning Over Paths in Knowledge Bases using Reinforcement Learning ICLR 2018 Investigating the Working of Text Classifiers COLING 2018 Nonparametric Density Estimation under Adversarial Losses NeurIPS 2018 Open Domain Question Answering Using Early Fusion of Knowledge Bases and Text EMNLP 2018 Transformation Autoregressive Networks ICML 2018 Canopy Fast Sampling with Cover Trees ICML 2017 Deep Sets NeurIPS 2017 Latent LSTM Allocation: Joint Clustering and Non-Linear Dynamic Modeling of Sequence Data ICML 2017 Question Answering on Knowledge Bases and Text using Universal Schema and Memory Networks ACL 2017 Exponential Stochastic Cellular Automata for Massively Parallel Inference AISTATS 2016 Gaussian LDA for Topic Models with Word Embeddings ACL 2015 Preferential Attachment in Graphs with Affinities AISTATS 2015