Papers › Rethinking Neural Operations for Diverse Tasks

Rethinking Neural Operations for Diverse Tasks

29 Mar 2021NeurIPS 2021 12arXiv:2103.15798archive 2025-07-28

Nicholas Roberts, Mikhail Khodak, Tri Dao, Liam Li, Christopher Ré, Ameet Talwalkar

An important goal of AutoML is to automate-away the design of neural networks on new tasks in under-explored domains. Motivated by this goal, we study the problem of enabling users to discover the right neural operations given data from their specific domain. We introduce a search space of operations called XD-Operations that mimic the inductive bias of standard multi-channel convolutions while being much more expressive: we prove that it includes many named operations across multiple application areas. Starting with any standard backbone such as ResNet, we show how to transform it into a search space over XD-operations and how to traverse the space using a simple weight-sharing scheme. On a diverse set of tasks -- solving PDEs, distance prediction for protein folding, and music modeling -- our approach consistently yields models with lower error than baseline networks and often even lower error than expert-designed domain-specific approaches.

PaperPDFConference PDFCodeCode Syntology ran

In Syntology Open this paper in Syntology's Atlas, the map of the papers in Syntology's graph and their citations.

For agents, Syntology's MCP tool lists every function and class Syntology harvested from this paper and whether it ran (how to connect): get_harvested_code_for_paper(arxiv_id="2103.15798")

Code

Syntology Ran 18 of 34 code samples harvested from 3 repositories linked to this paper; 16 have no recorded run. Of those that ran: 3 ran · honoured contract; 4 ran · our draft was wrong; 1 ran · fixture could not drive it; 10 ran with no contract checked.

By repository: official repository: 33 samples from 2 repositories, 18 ran; found in paper text by Syntology: 1 sample from 1 repository, 0 ran. The run record, sample by sample. “Ran” means executed on a synthesized input, not that the code is correct or reproduces the paper.

mkhodak/relax officialmentioned in papermentioned on GitHubpytorch report
nick11roberts/XD officialmentioned in papermentioned on GitHubpytorchApache-2.0 report
sjunhongshen/dash mentioned on GitHubpytorch report

Repository list and official/mentioned flags are the archive's, frozen 2025-07-28. Reachability, where shown, is from one Syntology probe window (2026-09-16 to 2026-09-18); repositories not probed show nothing. GitHub stars are not tracked.

Code Syntology ran Syntology

34 samples harvested; 18 ran; 3 honoured the contract we drafted; 16 have no recorded run. Read from Syntology's graph 2026-09-24; that is when this build read the record, not when the samples ran.

3ran · honoured contract
4ran · our draft was wrong
1ran · fixture could not drive it
10ran
16unverified

Licence: 0 of the 34 samples are pointer only, meaning Syntology does not serve that copy's text. This page shows no code text for any sample; each one links to its file in the repository.

Harvested from 3 repositories linked to this paper, official or community; each sample names its own and says which. “Ran” means the sample executed on a synthesized input. It does not mean the output is correct, and nothing here reproduces the paper's results. “Honoured” and “violated” refer to a contract Syntology drafted from the code itself; “our draft was wrong” and “fixture could not drive it” are failures of Syntology's instrument, not of the code.

Each sample ends with its code_sha256, Syntology's identity for that exact code. An agent fetches the stored sample with Syntology's MCP tool get_code(code_sha256="…") (how to connect); click an identity to copy that call.

Repository labels, per sample. official repository: The archive marks this repository official for the paper. named in the paper: The archive records that the paper mentions this repository; it is not marked official. community (archive-listed): In the archive's code links for this paper, not marked official and not recorded as mentioned in the paper. found in paper text by Syntology: Syntology found this repository in the paper's own text; whether it is the authors' implementation is not asserted. community: Not in the archive's code links for this paper; a community repository Syntology harvested. Samples from a repository marked official are listed first. Licence labels name the repository's licence as recorded at harvest. “Pointer only” means Syntology does not serve that copy's text, for one of four reasons: no licence file was found; the licence was not identified; the licence is recorded as permissive but that copy's record is not marked cleared; or the licence is outside the permissive list Syntology serves text under (MIT, Apache-2.0, BSD and similar). Some licences outside that list permit redistribution, such as WTFPL, and GPL-3.0 under its conditions; they are simply not on the list. Hover a licence label for the reason. File links open the file on GitHub at the default branch, which may have changed since the harvest.

AvgPool mkhodak/relax/relax/xd.py official repository ran · our draft was wrong MIT (permissive) · 2a31a6dcb06ddf63 · report
Complex2Real mkhodak/relax/relax/xd.py official repository ran · metamorphic tier: well formed fingerprinted MIT (permissive) · bec5cac219d82ded · report
ConvPerm mkhodak/relax/relax/xd.py official repository ran MIT (permissive) · ba89e07cd33d1453 · report
FNOPerm mkhodak/relax/relax/xd.py official repository ran fingerprinted MIT (permissive) · 746f20753fb4022f · report
Fourier mkhodak/relax/relax/xd.py official repository ran fingerprinted MIT (permissive) · 4b400ff00a487d94 · report
Pad mkhodak/relax/relax/xd.py official repository ran fingerprinted MIT (permissive) · 1dde65deadfc1f2d · report
TransposeFlip mkhodak/relax/relax/xd.py official repository ran fingerprinted MIT (permissive) · 7113106c9caa90dc · report
TransposePad mkhodak/relax/relax/xd.py official repository ran fingerprinted MIT (permissive) · 63dc4a7a7bfdd3ba · report
TransposePerm mkhodak/relax/relax/xd.py official repository ran fingerprinted MIT (permissive) · 6817170fe598f430 · report
TruncateHermitian mkhodak/relax/relax/xd.py official repository ran fingerprinted MIT (permissive) · df99f62038dc6b95 · report
Unpad mkhodak/relax/relax/xd.py official repository ran MIT (permissive) · 83fd4dada951a4ce · report
atrous_permutation mkhodak/relax/relax/xd.py official repository ran · honoured contract fingerprinted MIT (permissive) · 8597cc597b0fd17f · report
complex_dist nick11roberts/xd/chrysalis/xd.py official repository ran · our draft was wrong Apache-2.0 (permissive) · 9ec5d46f798ca92d · report
complex_norm nick11roberts/xd/chrysalis/xd.py official repository ran · our draft was wrong Apache-2.0 (permissive) · f99717b5f5887edd · report
int2tuple mkhodak/relax/relax/xd.py official repository ran · honoured contract fingerprinted MIT (permissive) · b65e5411454bdb5b · report
special_dist nick11roberts/xd/chrysalis/xd.py official repository ran · fixture could not drive it fingerprinted Apache-2.0 (permissive) · cc788ee6ee06b4df · report
transpose_permutation mkhodak/relax/relax/xd.py official repository ran · honoured contract fingerprinted MIT (permissive) · c4904be3964ec72b · report
truncate_freq mkhodak/relax/relax/xd.py official repository ran · our draft was wrong MIT (permissive) · 7378c3d22e68d903 · report
MaxPool nick11roberts/XD/chrysalis/darts.py official repository unverified Apache-2.0 (permissive) · 1ff4c96878cdae81 · report
SharedOperation mkhodak/relax/relax/xd.py official repository unverified MIT (permissive) · 94357aa818c939d9 · report
TensorProduct mkhodak/relax/relax/xd.py official repository unverified MIT (permissive) · da15dcc8176fc0a6 · report
XD mkhodak/relax/relax/xd.py official repository unverified MIT (permissive) · 0a723f7b3a1f7560 · report
accuracy nick11roberts/XD/DARTS/utils.py official repository unverified Apache-2.0 (permissive) · e93d9ddbfeb73ced · report
check_weight_norm nick11roberts/XD/chrysalis/chrysalis.py official repository unverified Apache-2.0 (permissive) · 2480ed0be135e57a · report
count_parameters_in_MB nick11roberts/XD/DARTS/utils.py official repository unverified Apache-2.0 (permissive) · 3e816289579cc9e9 · report
drop_path nick11roberts/XD/DARTS/utils.py official repository unverified Apache-2.0 (permissive) · 13c2719404e25b25 · report
fixed mkhodak/relax/relax/xd.py official repository unverified MIT (permissive) · 50254999b8752976 · report
get_module nick11roberts/XD/chrysalis/chrysalis.py official repository unverified Apache-2.0 (permissive) · e151df3b5424a3fb · report
multichannel_prod mkhodak/relax/relax/xd.py official repository unverified MIT (permissive) · 50200b49d8c95110 · report
pad2size mkhodak/relax/relax/xd.py official repository unverified MIT (permissive) · bc0b1e3f81223bd5 · report
resnet20 mkhodak/relax/examples/resnet/resnet.py official repository unverified MIT (permissive) · b3920efd145667ac · report
resnet32 mkhodak/relax/examples/resnet/resnet.py official repository unverified MIT (permissive) · ccbabd3d954fa0a8 · report
resnet44 mkhodak/relax/examples/resnet/resnet.py official repository unverified MIT (permissive) · 595fe4a7995fd021 · report
basic_fcn ba-lab/pdnet/models.py found in paper text by Syntology unverified Apache-2.0 (permissive) · 68a0988ac016a394 · report

Tasks

AutoMLImage ClassificationInductive BiasMusic ModelingNeural Architecture SearchProtein Folding

Results from the paper archive 2025-07-28

No leaderboard rows for this paper in the archive.

Methods

1x1 ConvolutionAverage PoolingBatch NormalizationBottleneck Residual BlockConvolutionGlobal Average PoolingKaiming InitializationMax PoolingReLUResidual BlockResidual Connection

Report a problem or propose a change · a person checks every report against the paper or source before anything changes; decisions are listed on /corrections