← Search

Kanishka Rao

28 accepted papers

2025

Little Is Enough: Boosting Privacy by Sharing Only Hard Labels in Federated Semi-Supervised Learning

AAAI 2025technical

In many critical applications, sensitive data is inherently distributed and cannot be centralized due to privacy concerns. A wide range of federated learning approaches have been proposed to train models locally at each client without sharing their sensitive data, typically by exchanging model param…

2024

How to Prompt Your Robot: A PromptBook for Manipulation Skills with Code as Policies

ICRA 2024poster

Large Language Models (LLMs) have demonstrated the ability to perform semantic reasoning, planning and write code for robotics tasks. However, most methods rely on pre-existing primitives (i.e. pick, open drawer) or similar examples of robot code alone, which heavily limits their scalability to new…

Cited by 30SourceScholar
2024

Learning to Learn Faster from Human Feedback with Language Model Predictive Control

RSS 2024poster

Large language models (LLMs) have been shown to exhibit a wide range of capabilities, such as writing robot code from language commands -- enabling non-experts to direct robot behaviors, modify them based on feedback, or compose them to perform new tasks. However, these capabilities (driven by in-co…

2024

Open X-Embodiment: Robotic Learning Datasets and RT-X Models : Open X-Embodiment Collaboration

ICRA 2024

Large, high-capacity models trained on diverse datasets have shown remarkable successes on efficiently tackling downstream applications. In domains from NLP to Computer Vision, this has led to a consolidation of pretrained models, with general pretrained backbones serving as a starting point for man

Cited by 910SourcecodeScholar
2024

Open X-Embodiment: Robotic Learning Datasets and RT-X Models : Open X-Embodiment Collaboration0

ICRA 2024poster

Large, high-capacity models trained on diverse datasets have shown remarkable successes on efficiently tackling downstream applications. In domains from NLP to Computer Vision, this has led to a consolidation of pretrained models, with general pretrained backbones serving as a starting point for man…

Cited by 259SourcecodeScholar
2024

RT-Trajectory: Robotic Task Generalization via Hindsight Trajectory Sketches

ICLR 2024spotlight

Generalization remains one of the most important desiderata for robust robot learning systems. While recently proposed approaches show promise in generalization to novel objects, semantic concepts, or visual distribution shifts, generalization to new tasks remains challenging. For example, a languag…

Cited by 53SourcePDFScholar
2024

SARA-RT: Scaling up Robotics Transformers with Self-Adaptive Robust Attention

ICRA 2024poster

We present Self-Adaptive Robust Attention for Robotics Transformers (SARA-RT): a new paradigm for addressing the emerging challenge of scaling up Robotics Transformers (RT) for on-robot deployment. SARA-RT relies on the new method of fine-tuning proposed by us, called up-training. It converts pre-tr…

Cited by 10SourceScholar
2023

Deep RL at Scale: Sorting Waste in Office Buildings with a Fleet of Mobile Manipulators

RSS 2023poster

We describe a system for deep reinforcement learning of robotic manipulation skills applied to a large-scale real-world task: sorting recyclables and trash in office buildings. Real-world deployment of deep RL policies requires not only effective training algorithms, but the ability to bootstrap rea…

Cited by 30SourcePDFScholar
2023

Large Language Models as General Pattern Machines

CoRL 2023poster

We observe that pre-trained large language models (LLMs) are capable of autoregressively completing complex token sequences—from arbitrary ones procedurally generated by probabilistic context-free grammars (PCFG), to more rich spatial patterns found in the Abstraction and Reasoning Corpus (ARC), a g…

Cited by 220SourceScholar
2023

Open-vocabulary Queryable Scene Representations for Real World Planning

ICRA 2023poster

Large language models (LLMs) have unlocked new capabilities of task planning from human instructions. However, prior attempts to apply LLMs to real-world robotic tasks are limited by the lack of grounding in the surrounding scene. In this paper, we develop NLMap, an open-vocabulary and queryable sce…

Cited by 209SourcecodeScholar
2023

Q-Transformer: Scalable Offline Reinforcement Learning via Autoregressive Q-Functions

CoRL 2023poster

In this work, we present a scalable reinforcement learning method for training multi-task policies from large offline datasets that can leverage both human demonstrations and autonomously collected data. Our method uses a Transformer to provide a scalable representation for Q-functions trained via o…

Cited by 106SourceScholar
2023

RT-1: Robotics Transformer for Real-World Control at Scale

RSS 2023poster

By transferring knowledge from large, diverse, task-agnostic datasets, modern machine learning models can solve specific downstream tasks either zero-shot or with small task-specific datasets to a high level of performance. While this capability has been demonstrated in other fields such as computer…

2023

RT-2: Vision-Language-Action Models Transfer Web Knowledge to Robotic Control

CoRL 2023poster

We study how vision-language models trained on Internet-scale data can be incorporated directly into end-to-end robotic control to boost generalization and enable emergent semantic reasoning. Our goal is to enable a single end-to-end trained model to both learn to map robot observations to actions a…

Cited by 1068SourceScholar
2023

Token Turing Machines

CVPR 2023poster

We propose Token Turing Machines (TTM), a sequential, autoregressive Transformer model with memory for real-world sequential visual understanding. Our model is inspired by the seminal Neural Turing Machine, and has an external memory consisting of a set of tokens which summarise the previous history…

2022

Do As I Can, Not As I Say: Grounding Language in Robotic Affordances

CoRL 2022oral

Large language models can encode a wealth of semantic knowledge about the world. Such knowledge could be extremely useful to robots aiming to act upon high-level, temporally extended instructions expressed in natural language. However, a significant weakness of language models is that they lack real…

Cited by 1747SourcecodeScholar
2021

RetinaGAN: An Object-aware Approach to Sim-to-Real Transfer

ICRA 2021poster

The success of deep reinforcement learning (RL) and imitation learning (IL) in vision-based robotic manipulation typically hinges on the expense of large scale data collection. With simulation, data to train a policy can be collected efficiently at scale, but the visual gap between sim and real make…

Cited by 115SourcecodeScholar
2020

RL-CycleGAN: Reinforcement Learning Aware Simulation-to-Real

CVPR 2020oral

Deep neural network based reinforcement learning (RL) can learn appropriate visual representations for complex tasks like vision-based robotic grasping without the need for manually engineering or prior learning a perception system. However, data for RL is collected via running an agent in the desir…

Cited by 241PDFScholar
2019

Off-Policy Evaluation via Off-Policy Classification

NeurIPS 2019poster

In this work, we consider the problem of model selection for deep reinforcement learning (RL) in real-world environments. Typically, the performance of deep RL algorithms is evaluated via on-policy interactions with the target environment. However, comparing models in a real-world environment for th…

Cited by 67SourcePDFScholar
2019

Streaming End-to-end Speech Recognition for Mobile Devices

ICASSP 2019accepted

End-to-end (E2E) models, which directly predict output character sequences given input speech, are good candidates for on-device speech recognition. E2E models, however, present numerous challenges: In order to be truly useful, such models must decode speech utterances in a streaming fashion, in rea…

Cited by 677SourceScholar
2018

Multi-Dialect Speech Recognition with a Single Sequence-to-Sequence Model

ICASSP 2018accepted

Sequence-to-sequence models provide a simple and elegant solution for building speech recognition systems by folding separate components of a typical system, namely acoustic (AM), pronunciation (PM) and language (LM) models into a single neural network. In this work, we look at one such sequence-to-…

Cited by 0SourceScholar
2018

Multilingual Speech Recognition with a Single End-to-End Model

ICASSP 2018accepted

Training a conventional automatic speech recognition (ASR) system to support multiple languages is challenging because the sub-word unit, lexicon and word inventories are typically language specific. In contrast, sequence-to-sequence models are well suited for multilingual ASR because they encapsula…

Cited by 292SourceScholar
2018

State-of-the-Art Speech Recognition with Sequence-to-Sequence Models

ICASSP 2018accepted

Attention-based encoder-decoder architectures such as Listen, Attend, and Spell (LAS), subsume the acoustic, pronunciation and language model components of a traditional automatic speech recognition (ASR) system into a single neural network. In previous work, we have shown that such architectures ar…

Cited by 0SourceScholar
2016

Personalized speech recognition on mobile devices

ICASSP 2016accepted

We describe a large vocabulary speech recognition system that is accurate, has low latency, and yet has a small enough memory and computational footprint to run faster than real-time on a Nexus 5 Android smartphone. We employ a quantized Long Short-Term Memory (LSTM) acoustic model trained with conn…

Cited by 0SourceScholar
2015

Grapheme-to-phoneme conversion using Long Short-Term Memory recurrent neural networks

ICASSP 2015accepted

Grapheme-to-phoneme (G2P) models are key components in speech recognition and text-to-speech systems as they describe how words are pronounced. We propose a G2P model based on a Long Short-Term Memory (LSTM) recurrent neural network (RNN). In contrast to traditional joint-sequence based G2P approach…

Cited by 0SourceScholar
2015

Learning acoustic frame labeling for speech recognition with recurrent neural networks

ICASSP 2015accepted

We explore alternative acoustic modeling techniques for large vocabulary speech recognition using Long Short-Term Memory recurrent neural networks. For an acoustic frame labeling task, we compare the conventional approach of cross-entropy (CE) training using fixed forced-alignments of frames and lab…

Cited by 0SourceScholar