atlas

Research map / Point clouds / Point cloud classification

Lie Neurons: Adjoint-Equivariant Neural Networks for Semisimple Lie Algebras

ICML 2024 · Point clouds › Point cloud classification · approach: Equivariant point cloud networks · 0 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes Lie Neurons, an adjoint-equivariant MLP-style network that takes semisimple Lie algebra elements (as R^K vectors) as input, generalizing Vector Neurons via the Killing form and adding Lie bracket and geometric channel mixing layers.

Problem

Building neural networks that are equivariant to the adjoint action of semisimple Lie groups on data living in Lie algebras, which existing equivariant networks (designed for vectors under left group action) do not handle.

Approach

Represent Lie algebra elements as coefficient vectors in R^K and use the negative Killing form (adjoint-invariant) for a ReLU-style nonlinearity and invariant layer, plus a Lie bracket nonlinearity (commutator with skip connection) and a geometric channel mixing layer built from outer products; with so(3) this reduces to Vector Neurons.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use Lie Neurons with the bracket nonlinearity instead of Vector NeuronsBCH regression on so(3) · BCH regression, Id and SO(3)-augmented testFrobenius Error: 0.617 -> 6.9\times 10^{-4} (Id); 0.617 -> 6.9\times 10^{-4} (SO(3))Helped
Add geometric channel mixing (ablation)Lie Neurons for free-rotating ISS dynamics · Multiple trajectories, 5 sec, Idtrajectory norm distance: 0.739 -> 0.005Helped
Add geometric channel mixingVector Neurons for ModelNet40 registration · ModelNet40 point cloud registrationAverage Registration Error (deg): 2.227 -> 1.934Helped
Use Lie Neurons instead of MLPFree-rotating ISS dynamics · Multiple trajectories, 5 sec, Idtrajectory norm distance: 0.428 -> 0.005Helped
Use adjoint-equivariant Lie Neurons instead of MLPPlatonic solid classification · Rotated poseaccuracy: 36.54\% -> 99.61\%Helped
Train MLP with augmentationPlatonic solid classification · Original poseaccuracy: 95.76\% -> 81.47\%Hurt
Use LN-LB instead of LN-LRsl(3) invariant regression · Id testMSE: 1.30 $\times$ 10$^{-3}$ -> 0.557Hurt
Use 2 LN-LB layers instead of 2 LN-LR layerssl(3) equivariant regression · Id testMSE: 0.213 -> $\mathbf{9.83\times10^{-10}}$Helped
Add the residual connection to LN-Bracket (ablation)LN-LB · sl(3) invariant regressionMSE: 4.838 -> 0.558Helped
Add the residual connection to LN-Bracket (ablation)LN-LB · sl(3) equivariant regressionMSE: 0.276 -> 9.6 \times 10^{-10}Helped
Add the residual connection to LN-Bracket (ablation)LN-LB · Platonic solid classificationaccuracy: 0.967 -> 0.986 (original); 0.959 -> 0.979 (rotated)Helped
Use Lie Neurons instead of MLPsp(4) invariant regression · SP(4) test augmentationMSE: 0.446 -> $\mathbf{2.70\times 10^{-4}}$Helped

Surprising findings

Limitations

Experimental setup

Datasets: ModelNet40 synthetic sp(4) invariant function data synthetic BCH data synthetic ISS free-rotation trajectories synthetic Platonic solid homography data

Benchmarks: sp(4) invariant function regression BCH formula regression ISS dynamics learning ModelNet40 point cloud registration sl(3) Platonic solid classification sl(3) invariant/equivariant function regression

Metrics: mean squared error invariance error Frobenius error log error trajectory norm error (rad/s) registration error (deg) classification accuracy equivariance error

Baselines: MLP MLP Augmentation EMLP e3nn Vector Neurons Vector Neurons (Mixing) Neural ODE MLP

Builds on: Vector Neurons Killing form Neural ODE EMLP

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy