← Search

Lawson L.S. Wong

13 accepted papers

2025

Approximate Equivariance in Reinforcement Learning

AISTATS 2025poster

Equivariant neural networks have shown great success in reinforcement learning, improving sample efficiency and generalization when there is symmetry in the task. However, in many problems, only approximate symmetry is present, which makes imposing exact symmetry inappropriate. Recently, approximate…

Cited by 0SourcecodeScholar
2025

Two by Two: Learning Multi-Task Pairwise Objects Assembly for Generalizable Robot Manipulation

CVPR 2025poster

3D assembly tasks, such as furniture assembly and component fitting, play a crucial role in daily life and represent essential capabilities for future home robots. Existing benchmarks and datasets predominantly focus on assembling geometric fragments or factory parts, which fall short in addressing…

2023

Integrating Symmetry into Differentiable Planning with Steerable Convolutions

ICLR 2023poster

To achieve this, we draw inspiration from equivariant convolution networks and model the path planning problem as a set of signals over grids. We demonstrate that value iteration can be treated as a linear equivariant operator, which is effectively a steerable convolution. Building upon Value Iterat…

Cited by 14SourcePDFScholar
2023

Modeling Dynamics over Meshes with Gauge Equivariant Nonlinear Message Passing

NeurIPS 2023poster

Data over non-Euclidean manifolds, often discretized as surface meshes, naturally arise in computer graphics and biological and physical systems. In particular, solutions to partial differential equations (PDEs) over manifolds depend critically on the underlying geometry. While graph neural networks…

Cited by 1SourcePDFScholar
2023

One-shot Imitation Learning via Interaction Warping

CoRL 2023poster

Learning robot policies from few demonstrations is crucial in open-ended applications. We propose a new method, Interaction Warping, for one-shot learning SE(3) robotic manipulation policies. We infer the 3D mesh of each object in the environment using shape warping, a technique for aligning point c…

Cited by 13SourcecodeScholar
2023

Scaling up and Stabilizing Differentiable Planning with Implicit Differentiation

ICLR 2023poster

Differentiable planning promises end-to-end differentiability and adaptivity. However, an issue prevents it from scaling up to larger-scale problems: they need to differentiate through forward iteration layers to compute gradients, which couples forward computation and backpropagation and needs to b…

Cited by 7SourcePDFScholar
2023

The Surprising Effectiveness of Equivariant Models in Domains with Latent Symmetry

ICLR 2023top-25%

Extensive work has demonstrated that equivariant neural networks can significantly improve sample efficiency and generalization by enforcing an inductive bias in the network architecture. These applications typically assume that the domain symmetry is fully described by explicit transformations of t…

Cited by 34SourcePDFScholar
2022

Toward Compositional Generalization in Object-Oriented World Modeling

ICML 2022oral

Compositional generalization is a critical ability in learning and decision-making. We focus on the setting of reinforcement learning in object-oriented environments to study compositional generalization in world modeling. We (1) formalize the compositional generalization problem with an algebraic a…

Cited by 28SourcePDFScholar
2019

Multi-Object Search using Object-Oriented POMDPs

ICRA 2019poster

A core capability of robots is to reason about multiple objects under uncertainty. Partially Observable Markov Decision Processes (POMDPs) provide a means of reasoning under uncertainty for sequential decision making, but are computationally intractable in large domains. In this paper, we propose Ob…

Cited by 52SourceScholar