DeepDendrite: A GPU ComputingFramework for Detailed Neuron Models

DeepDendrite:面向精细神经元模型的GPU 计算框架

Simulate morphologically detailed neurons and train networks built from them on GPUs.DeepDendrite turns dendritic dependencies into parallel work without changing the required solve order.

在 GPU 上仿真具有精细形态的神经元,并训练由它们构成的网络。DeepDendrite 将树突依赖关系转化为并行任务,同时保持方程求解所需的运算顺序。

Keep the biological detail in the computation.

DeepDendrite is built for workloads where morphology, ion-channel dynamics, synaptic location, and spike timing are part of the model—not details to be averaged away.

01 · SIMULATE

Resolve signals across detailed dendritic trees.

Run multi-compartment models with heterogeneous mechanisms and explicit dendritic structure, from single cells to populations of morphologically distinct neurons.

  • Voltage and spike dynamics
  • Cell-type-specific morphologies
  • Explicit spine-level models
02 · LEARN

Build learning systems from detailed neurons.

Use detailed human pyramidal cells as computational units in a network, propagate teaching signals through dendrites, and evaluate learned responses on image-classification tasks.

  • Mini-batch training
  • Detailed-neuron hidden layers
  • Clean and adversarial evaluation

Parallelize the tree, preserve the dependencies.

Solving cable equations is not an unordered batch operation. During triangularization, children must be processed before their parent; back-substitution reverses that dependency. DHS finds independent compartments that can run together.

Explore the scheduling principle →
  1. 01
    TOPOLOGY

    Read the dependency tree

    Convert compartment connectivity into parent–child relationships and calculate node depth.

  2. 02
    SCHEDULE

    Group independent work

    Select the deepest available candidates and arrange them into dependency-safe parallel levels.

  3. 03
    EXECUTION

    Map levels to the GPU

    Execute multiple compartments within each neuron and organize data around the GPU memory hierarchy.

Use detail to ask questions simpler models cannot.

The published workflows show how the same framework supports fine-grained biophysical experiments and learning with morphologically detailed neural networks.

Designed around irregular neural computation.

DeepDendrite combines an optimal tree schedule with GPU-aware storage and a CoreNEURON-based simulation engine. The advantage comes from matching the runtime to the structure of the model.

01

Parallelism inside each neuron

DHS exposes work across independent branches instead of assigning an entire detailed cell to one serial thread.

02

Memory-aware data organization

Aligned compartment data and register reuse reduce expensive global-memory transactions during the solve.

03

Simulation and learning in one framework

The GPU runtime supports conventional detailed-neuron simulation alongside the published HPC-Net learning workflow.

PAPER-REPORTED EVIDENCEEquation solving · full-spine simulation · HPC-Net training

Performance depends on the model, hardware, thread allocation, and comparison platform. The benchmark page pairs every headline result with its original paper figure and experimental scope.

View benchmarks and paper figures →

Get DeepDendrite from source.

Clone the public repository with Git. DeepDendrite currently uses a source-based installation, with GPU and compiler requirements documented in the build guide.

Git · HTTPS
git clone https://github.com/pkuzyc/DeepDendrite.git
cd DeepDendrite
Apache-2.0Source buildGitHub repository ↗
NEXT-GENERATION RESEARCH PREVIEW

HelioX

GPU-native simulation and training for biophysically detailed networks.

HelioX complements the NEURON modeling ecosystem with a GPU-native backend designed around irregular dendritic trees, sparse spikes, continuous-time dynamics, and structure-aware learning.

View the ICML 2026 paper ↗Paper and presentation only · no source code shown

From NEURON models to a GPU-native learning runtime.

Model specification remains in the established NEURON workflow. After the model is frozen, HelioX binds its states to a runtime that coordinates simulation, gradient computation, and parameter updates on the GPU.

MODEL SPECIFICATIONNEURON

Morphology, mechanisms, connectivity, and event routing

FREEZE POINTStable state binding

Runtime identifiers connect model variables to GPU execution

EXECUTION + LEARNINGHelioX

Simulation, analytical gradients, and parameter updates

  1. 01

    Sparse spike processing

    Move only the event data required when rare spikes occur.

  2. 02

    Unified mechanisms and state

    Provide consistent mechanism instantiation and stable variable access.

  3. 03

    Concurrent ODE construction

    Use multiple streams while preserving within-step ordering.

  4. 04

    DHS parallel tree solving

    Map dependency-safe dendritic work to custom GPU execution.

  5. 05

    Analytical-gradient learning

    Bind structure-aware gradients directly to simulator states.

Public evidence from the paper.

The reported results pair simulation throughput with numerical fidelity, then extend the same runtime to whole-circuit fitting and deep biophysical learning.

SIMULATION1.61–2.25×

Faster than CoreNEURON

Across the four FP64 simulation benchmarks reported in Table 1 on the paper's RTX 5090 setup.

FIDELITY<10⁻⁸ mV

Maximum L5PC voltage error

Against NEURON in the reported comparison, with identical spike timing in the tested case.

WHOLE-CIRCUIT LEARNING5.2 GB

Peak memory for C. elegans fitting

Compared with 47.4 GB for the CoreNEURON-based baseline on the reported RTX 4090 task.

BIOPHYSICAL MLP11.94× / 4.33×

Training / testing versus Jaxley

For the reported three-layer MNIST biophysical MLP comparison in Table 4.

These values describe the paper's tested models, hardware, precision, and baselines; they are not universal performance guarantees.

Presented at ICML 2026.

HelioX: A GPU-Native Framework for Simulation and Training of Biophysically Detailed Networks

Junfeng Lu · Zijie Yu · Shaoyang Cui · Gan He · Ruiqin Xiong · Kai Du · Tiejun Huang

43rd International Conference on Machine Learning

This preview summarizes information already disclosed in the paper and presentation. It does not include source code, private APIs, or release details.

View online at ICML ↗

DEEPDENDRITE · OPEN-SOURCE RESEARCH SOFTWARE

Bring detailed neurons to the GPU.