← Search

Akshat Shrivastava

10 accepted papers

2024

LayerSkip: Enabling Early Exit Inference and Self-Speculative Decoding

ACL 2024long

We present LayerSkip, an end-to-end solution to speed-up inference of large language models (LLMs). First, during training we apply layer dropout, with low dropout rates for earlier layers and higher dropout rates for later layers, and an early exit loss where all transformer layers share the same e…

2024

PRoDeliberation: Parallel Robust Deliberation for End-to-End Spoken Language Understanding

EMNLP 2024finding

Spoken Language Understanding (SLU) is a critical component of voice assistants; it consists of converting speech to semantic parses for task execution. Previous works have explored end-to-end models to improve the quality and robustness of SLU models with Deliberation, however these models have rem…

Cited by 0SourcePDFScholar
2024

PrE-Text: Training Language Models on Private Federated Data in the Age of LLMs

ICML 2024oral

On-device training is currently the most common approach for training machine learning (ML) models on private, distributed user data. Despite this, on-device training has several drawbacks: (1) most user devices are too small to train large models on-device, (2) on-device training is communication-…

2024

Small But Funny: A Feedback-Driven Approach to Humor Distillation

ACL 2024long

The emergence of Large Language Models (LLMs) has brought to light promising language generation capabilities, particularly in performing tasks like complex reasoning and creative writing. Consequently, distillation through imitation of teacher responses has emerged as a popular technique to transfe…

Cited by 4SourcePDFScholar
2023

ICASSP 2023 Spoken Language Understanding Grand Challenge

ICASSP 2023accepted

Spoken language understanding (SLU) is a important field between the Speech and NLP community focused on converting a users’ speech utterance into an executable semantic parse. In order to facilitate open research in this space, we introduce the 1st Spoken Language Understanding challenge hosted at…

Cited by 0SourceScholar
2023

Introducing Semantics into Speech Encoders

ACL 2023long

Recent studies find existing self-supervised speech encoders contain primarily acoustic rather than semantic information. As a result, pipelined supervised automatic speech recognition (ASR) to large language model (LLM) systems achieve state-of-the-art results on semantic spoken language tasks by u…

Cited by 4SourcePDFScholar
2021

Better Fine-Tuning by Reducing Representational Collapse

ICLR 2021poster

Although widely adopted, existing approaches for fine-tuning pre-trained language models have been shown to be unstable across hyper-parameter settings, motivating recent work on trust region methods. In this paper, we present a simplified and efficient method rooted in trust region theory that repl…

2021

Muppet: Massive Multi-task Representations with Pre-Finetuning

EMNLP 2021main

We propose pre-finetuning, an additional large-scale learning stage between language model pre-training and fine-tuning. Pre-finetuning is massively multi-task learning (around 50 datasets, over 4.8 million total labeled examples), and is designed to encourage learning of representations that genera…

2021

Non-Autoregressive Semantic Parsing for Compositional Task-Oriented Dialog

NAACL 2021long

Semantic parsing using sequence-to-sequence models allows parsing of deeper representations compared to traditional word tagging based models. In spite of these advantages, widespread adoption of these models for real-time conversational use cases has been stymied by higher compute requirements and…

2021

Span Pointer Networks for Non-Autoregressive Task-Oriented Semantic Parsing

EMNLP 2021finding

An effective recipe for building seq2seq, non-autoregressive, task-oriented parsers to map utterances to semantic frames proceeds in three steps: encoding an utterance x, predicting a frame’s length |y|, and decoding a |y|-sized frame with utterance and ontology tokens. Though empirically strong, th…

Cited by 25SourcePDFScholar