Zero Coordinate Shift (ZCS): Operator Learning & Codes
- Zero Coordinate Shift (ZCS) is a set of mathematical techniques addressing both high-order differentiation in physics-informed operator learning and interference-free code sequence design in communications.
- In operator learning, ZCS collapses differentiation complexities by introducing a shared shift variable, dramatically reducing computational memory and execution time.
- In communications, ZCS underpins zero correlation zone (ZCZ) sequence design, ensuring multi-user signal transmissions remain robust and interference-free.
Zero Coordinate Shift (ZCS) refers to a class of algorithmic and mathematical constructions that, depending on research context, address either (1) optimally structured code sequences for communication with guaranteed zero correlation within specific shift zones, or (2) efficient automatic differentiation for operator learning in physics-informed deep learning. Despite sharing a name, these usages are distinct in scope and technical realization. The following entry details both interpretations as independently developed in the contemporary literature.
1. ZCS in Physics-Informed Operator Learning
Motivation and Background
Zero Coordinate Shift (ZCS) in the context of physics-informed operator learning addresses the scalability bottleneck in automatic differentiation (AD) when computing high-order spatial or temporal derivatives with respect to collocation coordinates in Physics-Informed Neural Operators (PINOs) such as DeepONets. Standard AD frameworks require making all sampled coordinates leaf nodes, resulting in a many-roots-many-leaves (_∞∞) problem; this approach causes computational and memory requirements to scale poorly with the number of input functions and collocation points (Leng et al., 2023).
Existing attempts to circumvent this include explicit function loops (computing separate graphs for each input function) and data vectorization (treating the entire field as a single vector), each incurring memory and compute blowup. ZCS introduces a shared infinitesimal shift variable (per spatial dimension), collapsing these differentiation complexities into substantially smaller computational graphs.
Mathematical Formulation
Let denote the network output for function at collocation point . Standard AD computes derivatives by making all coordinates leaves in the graph. ZCS introduces a scalar 0 so that
1
which enables the desired derivative to be obtained as
2
Here, only 3 is a leaf, shifting all reliance on the coordinate dimension to a single computational node.
To recover the full 4 tensor of derivatives efficiently, one introduces a root-collapsing weight tensor 5 and considers
6
The entries 7 are then available via 8, and the corresponding derivatives as mixed second derivatives 9. Higher-order derivatives and mixed products follow recursively by this formalism, reducing computation from an _∞∞ problem to a sequence of _11 and _∞1 operations suitable for reverse-mode AD.
Algorithmic Realization
The procedure, expressed in PyTorch pseudocode for two spatial dimensions 0, is as follows:
7
This method requires only three scalar-to-scalar (11) and one vector-to-scalar (∞1) AD operations, eliminating the need for loops or coordinate duplication.
Implementation Considerations
The ZCS methodology is implemented by subclassing DeepXDE’s operator modules, overriding coordinate handling to use shift scalars and compressing the AD graph. The approach treats 1 as a batch dimension, with coordinate tensors never duplicated across input functions. The bulk memory and compute thus remain comparable to 2 PINN instances, with the only overhead coming from a small number of scalars and possibly the 3 tensor.
Empirical Performance
Extensive benchmarking on representative PDE problems (reaction-diffusion, Burgers, Kirchhoff–Love plates, Stokes) demonstrates that ZCS reduces computational graph memory consumption and wall time by an order of magnitude compared to previous approaches. For example, in a Burgers equation with 4, ZCS achieves 0.20 GB graph memory, 0.36 GB peak memory, and 15 s per 1000 batches, versus 7.8 GB, 7.9 GB, and 316 s for function loop methods (Leng et al., 2023). Relative errors are essentially unchanged between methods.
A summary table is provided:
| Problem (M,N,P) | Method | Graph Mem (GB) | Peak Mem (GB) | Time/1000 batches (s) | Rel. Error |
|---|---|---|---|---|---|
| Reaction–Diff (50,1000,2) | FuncLoop | 0.96 | 0.98 | 181 | 8.3% ±2.0 |
| DataVect | 0.97 | 1.46 | 24 | 9.5% ±2.5 | |
| ZCS | 0.02 | 0.05 | 10 | 8.2% ±2.0 | |
| Burgers (50,12800,2) | FuncLoop | 7.84 | 7.91 | 316 | 7.5% |
| DataVect | 7.73 | 11.40 | 221 | 7.2% | |
| ZCS | 0.20 | 0.36 | 15 | 7.1% ±0.5 | |
| Kirchhoff–Love (36,10000,4) | FuncLoop | 77.6 | 77.6 | 4081 | 27.3% |
| DataVect | – | – | – | – | |
| ZCS | 2.36 | 3.30 | 144 | 26.9% ±0.5 | |
| Stokes (50,5000,2) | FuncLoop | 74.4 | 78.6 | 4253 | 10.3% |
| DataVect | – | – | – | – | |
| ZCS | 1.99 | 3.30 | 147 | 10.4% ±0.6 |
A plausible implication is that ZCS facilitates the training of operator learning models at previously infeasible scales.
Scope and Limitations
ZCS is agnostic to the choice of network architecture, PDE type, or data sampling, and imposes no loss in training accuracy. It does not yield advantages for architectures leveraging explicit grid structure (e.g., Fourier Neural Operators or CNN-based PINNs) with optimized finite difference or FFT differentiators, where 5 stencils remain optimal. ZCS is formulated for reverse-mode AD, and its application to forward-mode scenarios is an area for potential extension (Leng et al., 2023).
2. ZCS and Zero Correlation Zone (ZCZ) Sequences
Definition and Theoretical Foundations
In communication theory, Zero Coordinate Shift (ZCS) is closely related to the zero correlation zone (ZCZ) property of sequences used for multi-carrier code division multiple access (MC-CDMA) and related systems. For two complex-valued sequences 6 and 7, the aperiodic cross-correlation is defined as
8
ZCZ of width 9 means 0 for all nonzero 1. For a collection of sequences, when auto- and cross-correlations satisfy this property, the set is a Z-complementary code set (ZCCS) (Shen et al., 2021).
A 2–ZCCS is a family 3 of 4 sequence sets, each 5 consisting of 6 sequences of length 7. The ZCCS definition requires
- 8 for 9 (sum auto-correlation vanishes outside zero shift).
- 0 for all 1 and 2 (mutual orthogonality over the zone).
The Feng–Fan–Zhou bound asserts 3; equality means optimality.
Construction via Extended Boolean Functions
Optimal ZCCS constructions are achieved using extended Boolean functions (EBF) 4:
5
with 6, 7, 8, 9 a permutation of 0, 1, 2.
Define
3
for 4 (in base 5), 6.
Partitioning the 7 sequences into 8 sets 9 yields an optimal 0–ZCCS: correlation vanishes up to shifts 1 (proof by generalized Golay argument and monomial analysis) (Shen et al., 2021).
A concrete illustration: for 2, 3, 4, 5, one recovers a 6–ZCCS by explicit computation.
Practical Implications
ZCZ sequences constructed this way are critical for asynchronous MC-CDMA, enabling multiple users to transmit signals robustly with delay offsets up to 7 without multi-user or multi-path interference. The practical trade-off—8—permits system design flexibility by adjusting sequence length 9, flock size 0, and zone width 1. The 2-ary construction generalizes beyond binary codes, supporting large, optimal ZCCS for high-capacity networks (Shen et al., 2021).
3. Applications in Deep Learning and Communications
In physics-informed deep learning, ZCS enables efficient training of neural operators for PDEs, with empirical evidence indicating scalability to previously unreachable sizes for operator dimensions 3 and collocation points 4, without increased training errors (Leng et al., 2023).
In wireless communications, ZCS in the ZCCS/ZCZ sense is foundational to MC-CDMA and analogous systems, supporting robust, interference-free multi-user transmission and flexible protocol design (Shen et al., 2021).
4. Comparative Summary and Terminological Distinctions
ZCS, as developed independently in operator learning and communications code design, addresses orthogonal concerns: backpropagation efficiency in the former, and interference-robust sequence structure in the latter. In operator learning, ZCS is a differentiable shift variable trick; in sequence design, it is a condition on correlation functions. Each is grounded in rigorous mathematical analysis and has led to substantial advances in its respective field.
5. Limitations and Prospective Extensions
In physics-informed operator learning, ZCS is not advantageous for grid-based models amenable to 5 stencil or FFT-based differentiation, and is primarily realized for reverse-mode AD systems. Anticipated extensions include porting to forward-mode AD in frameworks such as JAX or Julia for further performance improvement at high derivative orders.
In code design, ZCS/ZCZ constructions hinge on the algebraic constraints of the employed EBFs, and while highly flexible in 6-ary settings, remain subject to combinatorial existence bounds (Feng–Fan–Zhou).
6. References
- "Zero Coordinate Shift: Whetted Automatic Differentiation for Physics-informed Operator Learning" (Leng et al., 2023)
- "New Construction of Z-Complementary Code Sets and Mutually Orthogonal Complementary Sequence Sets" (Shen et al., 2021)