Papers by Rahul Gupta
Copied to clipboard
| Challenge: | Unlearning aims to remove copyrighted, sensitive, or private content from large language models without a full retraining. |
| Approach: | They propose a multi-task unlearning benchmark LUME that unlearns short novels, biographies and public biographie . |
| Outcome: | The proposed benchmark unlearns short novels, biographies and public biographie . it also releases fine-tuned models with 1B and 7B parameter sizes as targets . |
Copied to clipboard
| Challenge: | Recent studies have explored using LLMs for efficient data collection. |
| Approach: | They propose a method that takes into account the characteristics of the desired dataset and monitors the status of the generated data. |
| Outcome: | The proposed method improves safety and quality of three representative large language models against safety issues without sacrificing model utility. |
Copied to clipboard
| Challenge: | Existing tables models require linearization of the table structure, where row or column order is encoded as an unwanted bias. |
| Approach: | They propose a robust and structurally aware table-text encoding architecture TableFormer where tabular structural biases are incorporated completely through learnable attention biase. |
| Outcome: | The proposed architecture outperforms strong baselines on SQA, WTQ and TabFact table reasoning datasets and achieves state-of-the-art performance on SQ. |
Copied to clipboard
| Challenge: | Semantic parsing (SP) is a core component of modern virtual assistants like Google Assistant and Amazon Alexa. |
| Approach: | They propose a non-autoregressive (NAR) semantic parser that introduces intent conditioning on the decoder. |
| Outcome: | The proposed model reduces inference latency while maintaining competitive parsing quality. |
Copied to clipboard
| Challenge: | Large Language Models (LLMs) perform well on reasoning benchmarks but often fail when inputs alter slightly, raising concerns about overreliance on memorization. |
| Approach: | They propose a framework for Source-aware Token-level Identification of Memorization which attributes each token in a reasoning chain to one of multiple memorization sources based on their statistical co-occurrence with the token in the pretraining corpus. |
| Outcome: | The proposed framework attributes each token in a reasoning chain to one of multiple memorization sources based on their statistical co-occurrence with the token in the pretraining corpus. |
Copied to clipboard
| Challenge: | Large Language Models (LLMs) have a tendency to devolve into toxic degeneration . model may classify prompts as toxic or non-toxic and categorically refuse to respond to those deemed toxic. |
| Approach: | They propose a mechanism for LLM detoxification by labeling acceptable and unacceptable examples and including a corresponding acceptable rewrite with every unacceptable example. |
| Outcome: | The proposed model improves on the baseline model and shows that it detects and rewrites toxic and harmful examples. |
Copied to clipboard
| Challenge: | Reward modeling in large language models is susceptible to reward hacking . flawed reward signals often lead to outputs that optimize for spurious correlates . |
| Approach: | They propose a new approach that generates dynamic, context-relevant criteria to ground the reward model prior to producing reward scores. |
| Outcome: | The proposed approach generates dynamic, context-relevant criteria to ground the model prior to producing reward scores. |
Copied to clipboard
| Challenge: | Recent work shows that fine-tuning on insecure code data can trigger an emergent misalignment (EMA) phenomenon . |
| Approach: | They extend their study by demonstrating that EMA can arise from narrow refusal unlearning . they perform refusal unLearning on Cybersecurity and Safety concept and evaluate EMA . |
| Outcome: | The proposed model can generate malicious responses even to unrelated prompts . the proposed model is able to restore alignment across the affected domains while having lower refusal rates. |
Copied to clipboard
| Challenge: | Existing methods to identify code-mixed text are difficult to scale effectively and efficiently on multi-sentential data. |
| Approach: | They propose to identify multi-sentential code-mixed text (MCT) from multilingual articles using a token-level language-aware pipeline. |
| Outcome: | The proposed dataset includes 67k articles with 85k identified Hinglish MCTs. |
Copied to clipboard
| Challenge: | Existing methods to reduce toxic generation in large language models are not fully understood. |
| Approach: | They propose to understand the mechanisms that drive toxic generation in large language models by using memory localization to reduce toxic generation. |
| Outcome: | The proposed method reduces toxic generation from 62.86% to 28.61%, but it also improves generation quality. |
Copied to clipboard
| Challenge: | Current systems that focus on standard American English are not dialect invariant . current systems focus on a single dialect, which results in performance discrepancies . |
| Approach: | They propose a resource for evaluating and achieving English dialect invariance . they stress test question answering, machine translation, and semantic parsing . |
| Outcome: | The proposed system is based on a rule-based translation system spanning 50 English dialects and 189 unique linguistic features. |
Copied to clipboard
| Challenge: | Large language models (LLMs) have shown impressive reasoning ability, but many downstream reasoning tasks focus on performance-wise evaluation. |
| Approach: | They define and assess the Self-Contra rate across three datasets and delve into finer-grained categories of Self-contra reasoning. |
| Outcome: | The proposed model can detect self-contra reasoning with a 52.2% F1 score, much lower than for humans. |
Copied to clipboard
| Challenge: | Continual Federated Learning (CFL) combines decentralized learning with continuous learning . ubiquity of personal devices with a network connection offers rich source of data for learning problems . |
| Approach: | They propose to combine decentralized learning with a continuous learning approach . they propose to coordinate gradient-based replay sample selection across clients . |
| Outcome: | The proposed method shows gains early in the low replay size regime, when the budget for storing past data is small. |
Copied to clipboard
| Challenge: | Existing studies do not consider variance change due to metric model errors, which can lead to wrong conclusions. |
| Approach: | They establish the mathematical foundation of significance testing for model-based metrics . they show that metric errors can change the conclusions in certain experiments . |
| Outcome: | The proposed method can be used to derive accurate conclusions using model evaluations. |
Copied to clipboard
| Challenge: | Recent natural language processing systems use large language models as the backbone . however, societal biases are encoded in these models and transferred to downstream applications . |
| Approach: | They propose to use two categories to measure fairness in natural language processing tasks . they find intrinsic and extrinsic metrics do not correlate in their original setting . |
| Outcome: | The proposed metrics do not correlate in their original setting, the authors show . they find that they are not accurate when correcting for metric misalignments and noise . |
Copied to clipboard
| Challenge: | Recent studies have looked into the ability of large language models in various benchmark tasks, including question generation, reading comprehension, multilingual and etc. However, few studies investigate the controllability of large languages. |
| Approach: | They propose to compare large language models with state-of-the-start finetuned smaller models to find that large language model controls are comparable to smaller models. |
| Outcome: | The proposed model can meet hard constraints and perform better than state-of-the-art models. |
Copied to clipboard
| Challenge: | PRESTO dataset contains 550K contextual multilingual conversations between humans and virtual assistants. |
| Approach: | They propose to use a dataset of 550K contextual multilingual conversations between humans and virtual assistants to study some of the more challenging aspects of parsing realistic conversations. |
| Outcome: | The dataset contains 550K contextual conversations between humans and virtual assistants. |
Copied to clipboard
| Challenge: | Large language models generate biased responses where opinions of certain groups and populations are underrepresented. |
| Approach: | They propose a data-driven notion of persona that allows for a more nuanced understanding of different (latent) social groups present in the population. |
| Outcome: | The proposed method improves model steerability by 57% over baselines. |
Copied to clipboard
| Challenge: | Recent advances in Large language models have enabled them to hold free form conversations over multiple turns, but they exhibit a tendency to make unfounded and incorrect statements, commonly labeled as hallucinations. |
| Approach: | They propose a framework to test large language models for hallucinations using automatically generated INValId questions. |
| Outcome: | The proposed framework is based on a testbed of automatically generated INValId questions to evaluate large language models for hallucinations. |
Copied to clipboard
| Challenge: | Using a sequence-level constraint, we regularize the LLMtraining by penalizing the KL divergence between the desired output distribution and the LRM’s posterior. |
| Approach: | They propose a constraint learning schema forfine-tuning Large Language Models with attribute control by penalizing the KL divergence be-tween the desired output distribution and the LLM's posterior. |
| Outcome: | The proposed approach improves the performance of large language models while enhancing their utility and generation quality. |
Copied to clipboard
| Challenge: | Existing approaches for multi-objective Reinforcement Learning (RL) are difficult due to plurality of preferences and applications. |
| Approach: | They propose a framework for finetuning language models on multiple objectives using conditional language policy. |
| Outcome: | The proposed framework outperforms and Pareto-dominates existing approaches for multi-objective Reinforcement Learning (RL) it does not require training or maintaining multiple models to achieve different trade-offs between the objectives. |
Copied to clipboard
| Challenge: | Conventional unlearning approaches forget all tokens in a target document, including common tokens that carry general knowledge. |
| Approach: | They propose a method that identifies a critical subset of tokens within the forgetting set that is relevant to the unwanted information and unlearns only those tokens. |
| Outcome: | Experiments on two benchmarks and six baseline unlearning algorithms show that selective unlearning achieves effective unlearning on the targeted forget data. |
Copied to clipboard
| Challenge: | Recent work has evaluated the vulnerabilities of large generative models, such as DALL-E, ChatGPT, and GPT-4. |
| Approach: | They propose an automatic red teaming framework that evaluates a given black-box model and exposes its vulnerabilities against unsafe and inappropriate content generation. |
| Outcome: | The proposed framework evaluates a given black-box model and exposes its vulnerabilities against unsafe and inappropriate content generation. |
Copied to clipboard
| Challenge: | Large language models (LLMs) have demonstrated remarkable performance across a wide range of industrial applications. |
| Approach: | They propose two techniques for training and deploying small language models that deliver high performance for a variety of industry use cases. |
| Outcome: | The proposed techniques retain much of the quality of larger models while reducing training/serving costs and latency. |
Copied to clipboard
| Challenge: | a recent study documented the harmful limitations of gender binary-centric large language models . data scarcity is a known culprit, but the precise mechanisms through which scarcity affects this behavior remain underexplored. |
| Approach: | They propose to use BPE tokenization to enforce consistent tokenization across gendered pronouns to improve neopronoun proficiency. |
| Outcome: | The proposed methods outperform finetuning with standard BPE, and improve neopronoun proficiency. |
Copied to clipboard
| Challenge: | Increasing concerns and regulations about data privacy necessitate the study of privacy-preserving, decentralized learning methods for natural language processing tasks. |
| Approach: | They propose a framework for evaluating federated learning methods on four different tasks . they propose federation between Transformer-based language models and FL methods . |
| Outcome: | The proposed framework compares FL methods on four different tasks under non-IID partitioning strategies. |
Copied to clipboard
| Challenge: | Existing methods to reduce disparities in model outcomes have focused on data augmentation, debiasing model embeddings, or adding fairness-based optimization objectives during training. |
| Approach: | They propose to use certified word substitution robustness methods to improve equality of odds and equality of opportunity on multiple text classification tasks. |
| Outcome: | The proposed methods improve equality of odds and equality of opportunity on multiple text classification tasks. |
Copied to clipboard
| Challenge: | Language models excel at generating coherent text, but can be biased in multiple ways, including the unfounded association of male and female genders with gender-neutral professions. |
| Approach: | They propose to modify teacher probabilities and augment the training set to learn a fair model during knowledge distillation by modifying teacher probability and augmenting the training sets. |
| Outcome: | The proposed approach reduces gender disparity in open-ended text generated from the distilled and finetuned models with only a minor compromise in utility. |
Copied to clipboard
| Challenge: | Large Language Models (LLMs) are evolving rapidly on code generation tasks. |
| Approach: | They propose to automate the vulnerability code benchmark creation with iterative auto validation. |
| Outcome: | The proposed benchmark covers 232 CWE categories across C/C++, Java, and Python languages. |
Copied to clipboard
| Challenge: | Existing fairness metrics are not yet available to measure the fairness of language processing systems. |
| Approach: | They propose a new metric which measures fairness in machine learning models based on the model’s prediction sensitivity to perturbations in input features. |
| Outcome: | The proposed metric can be linked with a specific notion of group fairness and individual fairness, and correlates well with humans’ perception of fairness. |
Copied to clipboard
| Challenge: | Recent literature has focused on Model Inversion Attacks (ModIvA) that can extract training data from model parameters. |
| Approach: | They propose an attack that extracts canaries from NLU training data and reconstructs them using non-sensitive tokens. |
| Outcome: | The proposed attack can reconstruct a four digit code in the training dataset with a probability of 0.5 in its best configuration. |
Copied to clipboard
| Challenge: | Existing red-teaming approaches focus on policy-level weaknesses, but they overlook systemic weaknesses . aRES exploits dual-targeting weaknesses in both the core LLM and the RM simultaneously. |
| Approach: | a new framework uncovers weaknesses in both the core and the reward models simultaneously . a "Safety Mentor" generates semantically coherent adversarial prompts . |
| Outcome: | ARES uncovers weaknesses in both the core LLM and the RM simultaneously . it fine-tunes the LM to detect harmful content, then optimizes the core model . |
Copied to clipboard
| Challenge: | Recent advances in large language models (LLMs) have offered a strong potential for natural language systems to process informal language. |
| Approach: | They propose to use movie subtitles to evaluate slang in large language models . they find that smaller LLMs finetuned on the dataset achieve comparable performance . |
| Outcome: | The proposed dataset can be used to evaluate LLMs on slang detection and identification of regional and historical sources for interpretive insights. |
Copied to clipboard
| Challenge: | ambiguities can lead to misinterpretation and miscommunication in natural language . resolving ambiguity is notoriously hard for machines . |
| Approach: | They propose a framework to disambiguate prompts given to generative models by soliciting clarifications from the end user. |
| Outcome: | The proposed framework generates more faithful images better aligned with user intention in the presence of ambiguities. |
Copied to clipboard
| Challenge: | Differential privacy is an important privacy concern when building statistical models on data containing sensitive information. |
| Approach: | They propose a utility-preserving differentially private text transformation algorithm using auto-encoders that can be used to transform text to offer robustness against attacks and produce transformations with high semantic quality. |
| Outcome: | The proposed model performs better against membership inference attacks while offering lower to no degradation in the utility of the underlying transformation process compared to baselines. |
Copied to clipboard
| Challenge: | Safety reasoning paradigms require high-quality policy-embedded chain-of-thought datasets . generating such data through human annotations is prohibitively expensive . |
| Approach: | They propose AIDSAFE: Agentic Iterative Deliberation for Safety Reasoning . AIDS AFE leverages multi-agent deliberation to iteratively expand reasoning on safety policies . |
| Outcome: | The proposed model improves policy adherence and reasoning quality while maintaining acceptable utility and over-refusal accuracy. |
Copied to clipboard
| Challenge: | Large Language Models memorize significant portions of training data, which poses privacy risk. |
| Approach: | They propose a prompt-tuning approach to control the extraction rates of memorized content in large language models. |
| Outcome: | The proposed techniques yield 9.3% increase in extraction rate compared to baseline model . the proposed defense achieves 97.7% reduction with a perplexity increase of 16.9% . |
Copied to clipboard
| Challenge: | Academic datasets are often static and contain data that is annotated all at once based on fixed annotation guidelines. |
| Approach: | They propose to build a single-task continuous learning dataset from an existing dataset and release it along with the code to the research community. |
| Outcome: | The proposed model is based on an existing dataset and released to the research community. |
Copied to clipboard
| Challenge: | Artificial Intelligence (AI) and Machine Learning (ML) systems are becoming more popular and are causing concerns over user privacy. |
| Approach: | They propose a method for training ML models using positive and negative user feedback and a framework to extract labels on edge to make FL viable. |
| Outcome: | The proposed method improves significantly over a self-training baseline, achieving performance closer to models trained with full supervision. |
Copied to clipboard
| Challenge: | a new approach to annotate live traffic is emerging to be cost-effective and efficient . manual data annotation is expensive and not preferred for meeting customer privacy expectations . |
| Approach: | They propose a targeted synthetic data generation technique by inserting tokens into a given semantic signature. |
| Outcome: | The proposed approach achieves the same accuracy as training with all available data on a voice assistant dataset. |
Copied to clipboard
| Challenge: | Large-scale, voice-based conversational assistants process each utterance through a multi-stage pipeline that includes wakeword detection, automatic speech recognition (ASR), natural language understanding (NLU), entity resolution, and textto-speech. |
| Approach: | They propose a method to identify customer implicitly satisfied with Alexa's responses by leveraging interpretations of model behavior. |
| Outcome: | The proposed approach produces statistically significant improvements in both offline and online tests. |
Copied to clipboard
| Challenge: | Existing methods to embed signatures by adjusting token selection preferences during text generation are highly sensitive to paraphrasing and synonyms. |
| Approach: | They propose a framework that embeds watermark signatures into the semantic structure of a sentence using Abstract Meaning Representation (AMR). |
| Outcome: | Empirical evaluation shows SWAN matches state-of-the-art detection performance on unaltered watermarked text while improving robustness against paraphrasing. |
Copied to clipboard
| Challenge: | Knowledge graphs (KGs) complement Large Language Models (LLMs) by providing reliable, structured, domain-specific, and up-to-date external knowledge. |
| Approach: | They propose a zero-shot reasoning algorithm that augments black-box LLMs with one or more KGs. |
| Outcome: | The proposed algorithm significantly improves performance on question answering and KG question answering tasks. |
Copied to clipboard
| Challenge: | Experimental results show that neural machine translation engines built via FL can be easily adapted when an FL-based aggregation is applied to fuse different domains. |
| Approach: | They propose to use federated learning to fuse mixed-domain translation models with a centralized aggregation to improve their performance. |
| Outcome: | The proposed model can be easily adapted to a mixed-domain translation model with slight modifications in the training process and perform on par with state-of-the-art training models. |
Copied to clipboard
| Challenge: | In academic research, natural language understanding tasks are typically defined by creating annotated datasets in which each utterance is encountered once. |
| Approach: | They propose a method that explicitly uses utterance frequency in training data to learn models that are more robust to unknown distributions. |
| Outcome: | The proposed approach shows up to 7.02% relative improvement over baselines on the tail data. |