← Search

Jiankai Sun

34 accepted papers

2026

GRAD-NAV++: Vision-Language Model Enabled Visual Drone Navigation With Gaussian Radiance Fields and Differentiable Dynamics

RA-L 2026

Autonomous drones capable of interpreting and executing high-level language instructions in unstructured environments remain a long-standing goal. Yet existing approaches are constrained by their dependence on hand-crafted skills, extensive parameter tuning, or computationally intensive models unsui

Cited by 12SourcecodeScholar
2026

Understanding the Mixture-of-Experts with Nadaraya-Watson Kernel

ICLR 2026poster

Mixture-of-Experts (MoE) has become a cornerstone in recent state-of-the-art large language models (LLMs). Traditionally, MoE relies on $\mathrm{Softmax}$ as the router score function to aggregate expert output, a designed choice that has persisted from the earliest MoE models to modern LLMs, and is…

Cited by 0SourceScholar
2025

ARCH: Hierarchical Hybrid Learning for Long-Horizon Contact-Rich Robotic Assembly

CoRL 2025poster

Generalizable long-horizon robotic assembly requires reasoning at multiple levels of abstraction. While end-to-end imitation learning (IL) is a promising approach, it typically requires large amounts of expert demonstration data and often struggles to achieve the high precision demanded by assembly…

Cited by 0SourceScholar
2025

DAPE V2: Process Attention Score as Feature Map for Length Extrapolation

ACL 2025long

The attention mechanism is a fundamental component of the Transformer model, contributing to interactions among distinct tokens. In general, the attention scores are determined simply by the key-query products. However, this work’s occasional trial (combining DAPE and NoPE) of including additional M…

2025

Decoder-Hybrid-Decoder Architecture for Efficient Reasoning with Long Generation

NeurIPS 2025poster

Recent advances in language modeling have demonstrated the effectiveness of State Space Models (SSMs) for efficient sequence modeling. While hybrid architectures such as Samba and the decoder-decoder architecture, YOCO, have shown promising performance gains over Transformers, prior works have not i…

Cited by 0SourcecodeScholar
2025

GRaD-Nav: Efficiently Learning Visual Drone Navigation with Gaussian Radiance Fields and Differentiable Dynamics

IROS 2025

Autonomous visual navigation is an essential element in robot autonomy. Reinforcement learning (RL) offers a promising policy training paradigm. However, existing RL methods suffer from high sample complexity, poor sim-to-real transfer, and limited runtime adaptability. These problems are particular

Cited by 7SourcecodeScholar
2025

NopeRoomGS: Indoor 3D Gaussian Splatting Optimization without Camera Pose Input

NeurIPS 2025poster

Recent advances in 3D Gaussian Splatting (3DGS) have enabled real-time, high-fidelity view synthesis, but remain critically dependent on camera poses estimated by Structure-from-Motion (SfM), which is notoriously unreliable in textureless indoor environments. To eliminate this dependency, recent pos…

Cited by 0SourceScholar
2025

ParticleFormer: A 3D Point Cloud World Model for Multi-Object, Multi-Material Robotic Manipulation

CoRL 2025poster

3D world models (i.e., learning-based 3D dynamics models) offer a promising approach to generalizable robotic manipulation by capturing the underlying physics of environment evolution conditioned on robot actions. However, existing 3D world models are primarily limited to single-material dynamics us…

Cited by 0SourcecodeScholar
2025

SIREN: Semantic, Initialization-Free Registration of Multi-Robot Gaussian Splatting Maps

CoRL 2025poster

We present SIREN for registration of multi-robot Gaussian Splatting (GSplat) maps, with zero access to camera poses, images, and inter-map transforms for initialization or fusion of local submaps. To realize these capabilities, SIREN harnesses the versatility and robustness of semantics in three cri…

Cited by 0SourceScholar
2025

SeqPO-SiMT: Sequential Policy Optimization for Simultaneous Machine Translation

ACL 2025finding

We present Sequential Policy Optimization for Simultaneous Machine Translation (SeqPO-SiMT), a new policy optimization framework that defines the simultaneous machine translation (SiMT) task as a sequential decision making problem, incorporating a tailored reward to enhance translation quality while…

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
2023

Conformal Prediction for Uncertainty-Aware Planning with Diffusion Dynamics Model

NeurIPS 2023poster

Robotic applications often involve working in environments that are uncertain, dynamic, and partially observable. Recently, diffusion models have been proposed for learning trajectory prediction models trained from expert demonstrations, which can be used for planning in robot tasks. Such models hav…

Cited by 43SourcePDFScholar
2023

DPAUC: Differentially Private AUC Computation in Federated Learning

AAAI 2023technical

Federated learning (FL) has gained significant attention recently as a privacy-enhancing tool to jointly train a machine learning model by multiple participants. The prior work on FL has mostly studied how to protect label privacy during model training. However, model evaluation in FL might also le…

2023

MimicPlay: Long-Horizon Imitation Learning by Watching Human Play

CoRL 2023oral

Imitation learning from human demonstrations is a promising paradigm for teaching robots manipulation skills in the real world. However, learning complex long-horizon tasks often requires an unattainable amount of demonstrations. To reduce the high data requirement, we resort to human play data - vi…

Cited by 187SourcecodeScholar
2023

NeRF-Loc: Transformer-Based Object Localization Within Neural Radiance Fields

RA-L 2023

Neural Radiance Fields (NeRFs) have become a widely-applied scene representation technique in recent years, showing advantages for robot navigation and manipulation tasks. To further advance the utility of NeRFs for robotics, we propose a transformer-based framework, <monospace xmlns:mml="http://www

Cited by 14SourceScholar
2023

Weak Proxies are Sufficient and Preferable for Fairness with Missing Sensitive Attributes

ICML 2023poster

Evaluating fairness can be challenging in practice because the sensitive attributes of data are often inaccessible due to privacy constraints. The go-to approach that the industry frequently adopts is using off-the-shelf proxy models to predict the missing sensitive attributes, e.g. Meta (Alao et al…

2022

BEHAVIOR-1K: A Benchmark for Embodied AI with 1,000 Everyday Activities and Realistic Simulation

CoRL 2022oral

We present BEHAVIOR-1K, a comprehensive simulation benchmark for human-centered robotics. BEHAVIOR-1K includes two components, guided and motivated by the results of an extensive survey on "what do you want robots to do for you?". The first is the definition of 1,000 everyday activities, grounded in…

Cited by 205SourceScholar
2022

Differentially private multi-party data release for linear regression

UAI 2022poster

Differentially Private (DP) data release is a promising technique to disseminate data without compromising the privacy of data subjects. However the majority of prior work has focused on scenarios where a single party owns all the data. In this paper we focus on the multi-party setting, where differ…

Cited by 4SourcePDFScholar
2022

Egocentric Human Trajectory Forecasting With a Wearable Camera and Multi-Modal Fusion

RA-L 2022

In this letter, we address the problem of forecasting the trajectory of an egocentric camera wearer (ego-person) in crowded spaces. The trajectory forecasting ability learned from the data of different camera wearers walking around in the real world can be transferred to assist visually impaired peo

Cited by 24SourcecodeScholar
2022

Label Leakage and Protection in Two-party Split Learning

ICLR 2022poster

Two-party split learning is a popular technique for learning a model across feature-partitioned data. In this work, we explore whether it is possible for one party to steal the private label information from the other party during split training, and whether there are methods that can protect agains…

2022

PlaTe: Visually-Grounded Planning With Transformers in Procedural Tasks

RA-L 2022

In this work, we study the problem of how to leverage instructional videos to facilitate the understanding of human decision-making processes, focusing on training a model with the ability to plan a goal-directed procedure from real-world videos. Learning structured and plannable state and action sp

Cited by 67SourceScholar
2021

Adversarial Inverse Reinforcement Learning With Self-Attention Dynamics Model

RA-L 2021

In many real-world applications where specifying a proper reward function is difficult, it is desirable to learn policies from expert demonstrations. Adversarial Inverse Reinforcement Learning (AIRL) is one of the most common approaches for learning from demonstrations. However, due to the stochasti

Cited by 32SourcecodeScholar
2020

Cross-View Semantic Segmentation for Sensing Surroundings

RA-L 2020

Sensing surroundings plays a crucial role in human spatial perception, as it extracts the spatial configuration of objects as well as the free space from the observations. To facilitate the robot perception with such a surrounding sensing capability, we introduce a novel visual task called Cross-vie

Cited by 317SourcecodeScholar
2020

EndCold: An End-to-End Framework for Cold Question Routing in Community Question Answering Services

IJCAI 2020poster

Routing newly posted questions (a.k.a cold questions) to potential answerers with suitable expertise in Community Question Answering sites (CQAs) is an important and challenging task. The existing methods either focus only on embedding the graph structural information and are less effective for newl…

2020

Learning a Decision Module by Imitating Driver’s Control Behaviors

CoRL 2020

Autonomous driving systems have a pipeline of perception, decision, planning, and control. The decision module processes information from the perception module and directs the execution of downstream planning and control modules. On the other hand, the recent success of deep learning suggests that t

2020

SegVoxelNet: Exploring Semantic Context and Depth-aware Features for 3D Vehicle Detection from Point Cloud

ICRA 2020poster

3D vehicle detection based on point cloud is a challenging task in real-world applications such as autonomous driving. Despite significant progress has been made, we observe two aspects to be further improved. First, the semantic context information in LiDAR is seldom explored in previous works, whi…

Cited by 78SourceScholar
2020

Transferable Active Grasping and Real Embodied Dataset

ICRA 2020poster

Grasping in cluttered scenes is challenging for robot vision systems, as detection accuracy can be hindered by partial occlusion of objects. We adopt a reinforcement learning (RL) framework and 3D vision architectures to search for feasible viewpoints for grasping by the use of hand-mounted RGB-D ca…

Cited by 26SourcecodeScholar