Zheyu Zhuang holding his camera

Hi, I’m

Zheyu

Zheyu Zhuang · Postdoctoral researcher at KTH

Robots that adapt.
And understand the task.

I study how robots learn efficiently, adapt to new situations, and follow human intent through language.

My work connects perception, imitation learning and real robots. I’m interested in how they understand long, multi-step tasks: what matters now, what comes next, and when to change course.

I work with Professor Danica Kragic at KTH. My PhD with Professor Robert Mahony at ANU focused on learning-based visual control.

Selected research

A selection of projects, from learning where to look to taking a skill into a new setting.

Attention for manipulation

Attention from action alone

Seeker CoRL 2026

Seeker learns where to look through action prediction alone, without spatial labels. The same attention guides zoom, background augmentation and 3D point filtering.

3×
real-robot success

60% versus 20% for the strongest baseline, averaged across lighting and background changes.

Predicted regions & masks · two camera views

Attention-guided training images

Three attention-guided appearance augmentations

New backgrounds at evaluation

Square insertion on a cereal-patterned tabletopBlock stacking on a leaf-patterned tabletop
Coffee-preparation point cloud before Seeker filtering Seeker-selected cup and gripper points, enlarged for clarity
Before filtering · workspace crop

Task-aware efficiency

A shallow CNN is enough —
when it knows where to look

FocusPool CoRL 2026

Task-conditioned attentionCoffee transfer · real robot
Recorded FocusPool attention overlay during a coffee-transfer task

Six recorded moments, looping through the task’s changing focus.

FocusPool selects shallow visual features according to the task’s progress. Knowing where to look makes a small encoder enough for effective control.

1.5%
of ResNet-18’s
encoder parameters

Using only the shallow L1 features.

The L2 model (5.8%) reaches comparable RVT-2 performance with roughly half the demonstrations: 50 versus 100 in the simulation comparison.

Skill transfer & spatial generalisation

Reusing skills in new settings

PALM RA-L 2026

Settings can look different overall, yet look similar near the gripper. PALM pairs this local visual similarity with proprioceptive representation alignment to reuse manipulation skills across workspace positions, camera viewpoints and robot embodiments.

No NEW demonstrations

Needed to transfer across positions, viewpoints and robot embodiments.

All publications

Research in robot learning, visual control and perception, newest first.

View listClose list
Semantic Segmentation from Limited Training Data A. Milan, T. Pham, K. Vijay, D. Morrison, A. W. Tow, L. Liu, J. Erskine, R. Grinover, A. Gurman, T. Hunn, N. Kelly-Boxall, D. Lee, M. McTaggart, G. Rallos, A. Razjigaev, T. Rowntree, T. Shen, R. Smith, S. Wade-McCue, Zheyu Zhuang, C. Lehnert, G. Lin, I. Reid, P. Corke, J. Leitner · ICRA · 2018

System integration

Real-world robot systems

I integrate hardware and software through ROS, connecting perception, sensor synchronisation, calibration and real-time control in complete robot systems.

GoferBot collaborating during furniture assembly, with human action recognition and visual servoing views

Human–robot collaboration

GoferBot

I co-developed GoferBot, a vision-based human–robot interaction system for collaborative furniture assembly. It combines human action recognition, visual servoing and handover behaviour so the robot can coordinate naturally with a person.

Cartman warehouse picking robotSemantic segmentation of warehouse objects for the Cartman project

Warehouse picking · ACRV team

Cartman

I contributed as a member of the ACRV team that built Cartman, combining custom hardware with robotic perception for warehouse picking. The system won the 2017 Amazon Robotics Challenge, and the team leader later spun this work out into Lyro Robotics.

Mentoring & service

Mentoring

I work with PhD researchers on questions, experiments and clear communication, with publications at CoRL, RA-L and ICRA.

Teaching

Six courses tutored at ANU, plus hands-on Robotic Vision Summer School workshops I designed and led. The workshop material later became part of teaching at Monash University.

Academic service

Associate Editor for IROS 2025 and 2026 and ICRA 2027. Reviewer for IEEE RA-L, IEEE T-RO, RSS, CoRL, ICLR, ICRA and IROS.

Outside the lab

Usually outside.
Often off-grid.

Attention from Action, for Action: Emergent Visual Bottlenecks for Policy Learning Attention learned from action alone, reused for zooming, background robustness and 3D point filtering
Enlarged Seeker regions of interest across two real-robot manipulation tasks
Localized Visual Feature Aggregation via Focus Pooling for Visuomotor Policies Task-progress-aware feature selection makes a shallow encoder enough Open full image ↗
Enlarged comparison of FocusPool, average pooling and spatial softmax using CNN feature maps
PALM: Enhanced Generalizability for Local Visuomotor Policies via Perception Alignment Visual and proprioceptive alignment for local manipulation
Enlarged PALM perception alignment figure showing changes in workspace position, camera viewpoint, and robot embodiment with a TCP-centric crop
RoboSaGA Real-robot toy manipulation · varied backgrounds, clutter and low light · 4× speed
Feature Extractor or Decision Maker: Rethinking the Role of Visual Encoders in Visuomotor Policies Same scene, different policy objective
Enlarged comparison of task-dependent encoder attention for reach-spam and reach-mug policies
MirrorDuo Consistent reflection of images, proprioception and 6-DoF actions
Enlarged MirrorDuo reflection-consistent visuomotor learning animation
Robot-centric Pooling Contrastive image–proprioception alignment for self–other distinction
Enlarged Robot-centric Pooling comparison of alignment scores and saliency maps