cifar10dvs_architecture.mmd

CIFAR10-DVS architecture diagram - 55 lines. View on GitHub (image-neuro/cifar10dvs_architecture.mmd).

Rendered diagram

Rendering diagram…

Source

graph TD
    INPUT["CIFAR10-DVS Input<br/><i>N × 16 × 2 × 32 × 32 DVS frames</i>"]

    subgraph CF["VGGSNN Convolutional Front-End"]
        direction LR
        C1["Conv2d 2→64<br/>BN + ReLU"]
        C2["Conv2d 64→128<br/>BN + ReLU + AvgPool + Dropout<br/><i>128 × 16 × 16</i>"]
        C3["Conv2d 128→256 + 256→256<br/>BN + ReLU + AvgPool + Dropout<br/><i>256 × 8 × 8</i>"]
        C4["Conv2d 256→512 + 512→512<br/>BN + ReLU + AvgPool + Dropout<br/><i>512 × 4 × 4</i>"]
        C5["Conv2d 512→512<br/>BN + ReLU<br/><i>512 × 4 × 4</i>"]
        C1 --> C2 --> C3 --> C4 --> C5
    end

    FL["Flatten + Linear + LayerNorm<br/><i>8192 → 256</i>"]
    DY["DyNED<br/>16 levels"]
    PU["Linear + LayerNorm<br/><i>256 → 1024</i>"]

    subgraph SL["cAdLIF Spiking Layers"]
        direction LR
        S1["Linear + BN + Delay + cAdLIF<br/>Drop 0.21<br/><i>1024 → 1024</i><br/>Surrogate: ArcTan"]
        S2["Linear + BN + Delay + cAdLIF<br/>Drop 0.21<br/><i>1024 → 1024</i><br/>Surrogate: ArcTan"]
        S3["Linear + BN + Delay + cAdLIF<br/>Drop 0.21<br/><i>1024 → 1024</i><br/>Surrogate: ArcTan"]
        S4["Readout: Linear + Leaky Integrator<br/><i>1024 → 10</i>"]
        S1 --> S2 --> S3 --> S4
    end

    TS["16 Time Steps per Sample"]
    TET["TET Loss<br/>(starts at epoch 50)"]
    CLASS["10 CIFAR10-DVS Classes"]

    INPUT --> CF
    CF --> FL
    FL --> DY
    DY --> PU
    PU --> SL
    TS -.- SL
    TET -.- SL
    SL --> CLASS

    classDef inputStyle fill:#27AE60,stroke:#1E8449,color:white,font-weight:bold,font-size:20px
    classDef convStyle fill:#3498DB,stroke:#2471A3,color:white,font-weight:bold,font-size:20px
    classDef projStyle fill:#8E44AD,stroke:#6C3483,color:white,font-weight:bold,font-size:20px
    classDef dynedStyle fill:#E67E22,stroke:#CA6F1E,color:white,font-weight:bold,font-size:20px
    classDef cadlifStyle fill:#C0392B,stroke:#922B21,color:white,font-weight:bold,font-size:20px
    classDef classStyle fill:#2C3E50,stroke:#1B2631,color:white,font-weight:bold,font-size:20px
    classDef noteStyle fill:#F5F5F5,stroke:#999,color:#333,font-style:italic,font-size:20px

    class INPUT inputStyle
    class C1,C2,C3,C4,C5 convStyle
    class FL,PU projStyle
    class DY dynedStyle
    class S1,S2,S3,S4 cadlifStyle
    class CLASS classStyle
    class TS,TET noteStyle