← Search

Sudipta Sengupta

7 accepted papers

2024

BASS: Batched Attention-optimized Speculative Sampling

ACL 2024findings

Speculative decoding has emerged as a powerful method to improve latency and throughput in hosting large language models. However, most existing implementations focus on generating a single sequence. Real-world generative AI applications often require multiple responses and how to perform speculativ…

2024

Bifurcated Attention for Single-Context Large-Batch Sampling

ICML 2024poster

In our study, we present bifurcated attention, a method developed for language model inference in single-context batch sampling contexts. This approach aims to reduce redundant memory IO costs, a significant factor in latency for high batch sizes and long context lengths. Bifurcated attention achiev…

Cited by 1SourcePDFScholar
2024

Synatra: Turning Indirect Knowledge into Direct Demonstrations for Digital Agents at Scale

NeurIPS 2024poster

LLMs can now act as autonomous agents that interact with digital environments and complete specific objectives (e.g., arranging an online meeting). However, accuracy is still far from satisfactory, partly due to a lack of large-scale, direct demonstrations for digital tasks. Obtaining supervised dat…

Cited by 19SourcePDFScholar
2023

A Static Evaluation of Code Completion by Large Language Models

ACL 2023industry

Large language models trained on code have shown great potential to increase productivity of software developers. Several execution-based benchmarks have been proposed to evaluate functional correctness of model-generated code on simple programming problems. Nevertheless, it is expensive to perform…

2023

Multi-lingual Evaluation of Code Generation Models

ICLR 2023top-25%

We present two new benchmarks, MBXP and Multilingual HumanEval, designed to evaluate code completion models in over 10 programming languages. These datasets are generated using a conversion framework that transpiles prompts and test cases from the original MBPP and HumanEval datasets into the corres…

2022

Improving Text-to-SQL Semantic Parsing with Fine-grained Query Understanding

EMNLP 2022industry

Most recent research on Text-to-SQL semantic parsing relies on either parser itself or simple heuristic based approach to understand natural language query (NLQ). When synthesizing a SQL query, there is no explicit semantic information of NLQ available to the parser which leads to undesirable genera…

Cited by 3SourcePDFScholar
2022

Neuro-Symbolic Language Modeling with Automaton-augmented Retrieval

ICML 2022spotlight

Retrieval-based language models (R-LM) model the probability of natural language text by combining a standard language model (LM) with examples retrieved from an external datastore at test time. While effective, a major bottleneck of using these models in practice is the computationally costly datas…