← Search

Yunfei Bai

16 accepted papers

2026

Learning Beyond Vision: Vision-Language Distillation and Edge-Aware Mix Diffusion in Semi-Supervised Semantic Segmentation

AAAI 2026technical

In semi-supervised semantic segmentation (SSSS), segmentation performance is heavily constrained by the quality of pseudo labels. However, prevalent pseudo-label optimization approaches rely on the model’s internal self-correction. When the model fails to recognize or adequately represent certain cl

Cited by 0SourcePDFScholar
2026

RECoRD: A Multi-Agent LLM Framework for Reverse Engineering Codebase to Relational Diagram

AAAI 2026technical

Understanding the behavior and logical structure of complex algorithms is a fundamental challenge in industrial systems. Recent advancements in large language models (LLMs) have demonstrated remarkable code understanding capabilities. However, their potential for reverse engineering algorithms into

Cited by 0SourcePDFScholar
2025

Stability Enhancement in Variable Morphing Multi-body AUVs for Underwater Structure Maintenance

IROS 2025

This paper presents a Variable Morphing Multi-Body AUVs (VMMAUVs) concept, designed for underwater structure maintenance. This robot is capable of dynamically adjusting their structure to adapt to varying operational scenarios. The study explores two key stability mechanisms: buoyancy adjustment and

Cited by 0SourceScholar
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

On Designing a Learning Robot: Improving Morphology for Enhanced Task Performance and Learning

IROS 2023poster

As robots become more prevalent, optimizing their design for better performance and efficiency is becoming increasingly important. However, current robot design practices overlook the impact of perception and design choices on a robot's learning capabilities. To address this gap, we propose a compre…

Cited by 1SourcecodeScholar
2023

Practical Visual Deep Imitation Learning via Task-Level Domain Consistency

ICRA 2023poster

Recent work in visual end-to-end learning for robotics has shown the promise of imitation learning across a variety of tasks. Such approaches are however expensive both because they require large amounts of real world data and rely on time-consuming real-world evaluations to identify the best model…

Cited by 3SourceScholar
2021

COCOI: Contact-aware Online Context Inference for Generalizable Non-planar Pushing

IROS 2021poster

General contact-rich manipulation problems are long-standing challenges in robotics due to the difficulty of understanding complicated contact physics. Deep reinforcement learning (RL) has shown great potential in solving robot manipulation tasks. However, existing RL policies have limited adaptabil…

Cited by 15SourcecodeScholar
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
2021

SimGAN: Hybrid Simulator Identification for Domain Adaptation via Adversarial Reinforcement Learning

ICRA 2021poster

As learning-based approaches progress towards automating robot controllers design, transferring learned policies to new domains with different dynamics (e.g. sim-to-real transfer) still demands manual effort. This paper introduces SimGAN, a framework to tackle domain adaptation by identifying a hybr…

Cited by 78SourcecodeScholar
2020

Online Learning of Object Representations by Appearance Space Feature Alignment

ICRA 2020poster

We propose a self-supervised approach for learning representations of objects from monocular videos and demonstrate it is particularly useful for robotics. The main contributions of this paper are: 1) a self-supervised model called Object-Contrastive Network (OCN) that can discover and disentangle o…

Cited by 15SourceScholar
2020

Watch, Try, Learn: Meta-Learning from Demonstrations and Rewards

ICLR 2020poster

Imitation learning allows agents to learn complex behaviors from demonstrations. However, learning a complex vision-based task may require an impractical number of demonstrations. Meta-imitation learning is a promising approach towards enabling agents to learn a new task from one or a few demonstrat…

Cited by 69SourceScholar
2018

Learning 6-DOF Grasping Interaction via Deep Geometry-Aware 3D Representations

ICRA 2018poster

This paper focuses on the problem of learning 6- DOF grasping with a parallel jaw gripper in simulation. Our key idea is constraining and regularizing grasping interaction learning through 3D geometry prediction. We introduce a deep geometry-aware grasping network (DGGN) that decomposes the learning…

Cited by 139SourceScholar
2018

Multi-Task Domain Adaptation for Deep Learning of Instance Grasping from Simulation

ICRA 2018poster

Learning-based approaches to robotic manipulation are limited by the scalability of data collection and accessibility of labels. In this paper, we present a multi-task domain adaptation framework for instance grasping in cluttered scenes by utilizing simulated robot experiments. Our neural network t…

Cited by 135SourceScholar
2018

Sim-to-Real: Learning Agile Locomotion For Quadruped Robots

RSS 2018poster

Designing agile locomotion for quadruped robots often requires extensive expertise and tedious manual tuning. In this paper, we present a system to automate this process by leveraging deep reinforcement learning techniques. Our system can learn quadruped locomotion from scratch using simple reward s…

Cited by 992SourcePDFScholar
2018

Using Simulation and Domain Adaptation to Improve Efficiency of Deep Robotic Grasping

ICRA 2018poster

Instrumenting and collecting annotated visual grasping datasets to train modern machine learning algorithms can be extremely time-consuming and expensive. An appealing alternative is to use off-the-shelf simulators to render synthetic data for which ground-truth annotations are generated automatical…

Cited by 827SourceScholar