equivariant-architecture-designer

Design neural network architectures respecting identified symmetry groups.

16|Updated Dec 28, 2025
One-click install
npx skills add https://github.com/Hongyu-yu/matsci-ai-skills --skill equivariant-architecture-designer-hongyu-yu
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: equivariant-architecture-designer
Source: https://github.com/Hongyu-yu/matsci-ai-skills/tree/main/skills/equivariant-architecture-designer
Command: npx skills add https://github.com/Hongyu-yu/matsci-ai-skills --skill equivariant-architecture-designer-hongyu-yu

SYSTEM DOCUMENTATION & REQUIREMENTS

What problem does it solve?

This skill helps practitioners design neural network architectures that respect identified symmetry groups, ensuring models are more data-efficient and robust to distribution shifts.

Core Features & Use Cases

  • Guided architecture selection for symmetry groups (e.g., Cn, SO(3), E(3)) and task types (classification, segmentation, regression).
  • Reference patterns & templates for equivariant layers, non-linearities, and pooling to preserve or exploit symmetry.
  • Implementation guidance including library choices (e.g., e3nn, escnn) and evaluation criteria to verify equivariance.

Quick Start

Provide a symmetry specification and requirements to generate an architecture plan.

Frequently Asked Questions about equivariant-architecture-designer

High-intent search queries and answers about installing and using this skill.

FAQPage Schema
How do I design a neural network architecture that respects SO(3) or E(3) symmetry groups?

To design an equivariant neural network respecting SO(3) or E(3) symmetry, provide a symmetry specification and task requirements to generate an architecture plan with appropriate layer patterns and non-linearities.

What is an equivariant neural network and when do I need to use symmetry-respecting architectures?

An equivariant neural network respects identified symmetry groups to ensure models remain data-efficient and robust to distribution shifts across images, point clouds, and graphs for tasks like classification or regression.

How do I implement equivariant layers and pooling in PyTorch using e3nn or escnn?

Implementation guidance for equivariant layers, non-linearities, and pooling in PyTorch is provided through library recommendations like e3nn and escnn, alongside evaluation criteria to verify equivariance.

Does this architecture selection tool support point clouds and graphs for segmentation tasks?

Yes, architecture selection supports point clouds and graphs across tasks such as classification, segmentation, and regression by applying reference patterns and templates for symmetry groups like Cn and E(3).

What's the best way to verify equivariance in a symmetry-respecting neural network?

The best way to verify equivariance is by applying the provided evaluation criteria to test that the identified symmetry groups are preserved through the network's equivariant layers and pooling operations.

Why does my neural network fail to generalize with distribution shifts on point cloud data?

Standard networks fail on distribution shifts because they lack symmetry constraints; designing architectures that respect identified symmetry groups like E(3) ensures models remain robust and data-efficient.