Skip to main content

A cerebellar-inspired prediction-correction engine for LLM Agents

Project description

Digital Cerebellum 数字小脑

A cerebellar-inspired prediction-correction engine for LLM Agents.

一个受生物小脑启发的轻量级预测-校正系统,作为大语言模型的"另一半大脑"。


What is this?

Current AI agents (ChatGPT, OpenClaw, LangChain, etc.) are all "cerebral cortex" — slow, expensive, flexible reasoning. But biological intelligence runs on two engines: the cerebral cortex and the cerebellum.

The cerebellum holds ~50% of all brain neurons. It doesn't think — it predicts, corrects, and automates. It's why you can catch a ball without calculating parabolas, why a pianist's fingers move faster than conscious thought.

Digital Cerebellum brings this missing half to AI agents:

  • < 10ms prediction latency (vs 1-10s for LLM)
  • Near-zero cost for routine decisions (vs $0.01+ per LLM call)
  • Online learning — gets better with every interaction
  • Uncertainty quantification — knows what it doesn't know

How it works

LLM Agent prepares a tool call
        │
        ▼
┌─ Digital Cerebellum ──────────────────────┐
│                                            │
│  Feature Encoder     (mossy fibres)        │
│       │                                    │
│  Pattern Separator   (granule cells, RFF)  │
│       │                                    │
│  K-Head Predictor    (Purkinje population) │
│       │                                    │
│  Decision Router     (deep cerebellar      │
│       │               nuclei)              │
│       ▼                                    │
│  High confidence → ALLOW  (fast path)      │
│  Low confidence  → ASK LLM (slow path)    │
│       │                                    │
│  Result → Error Signal → Online Learning   │
└────────────────────────────────────────────┘

Every biological component maps to a neuroscience-validated mechanism:

Biology Digital Reference
Granule cell layer Random Fourier Features Bhalla 2022, Frontiers Comp Neuro
Purkinje population K independent linear heads 2025 J.Neuroscience
Climbing fibres 3 error channels (SPE/TPE/RPE) 2025 Nature Communications
Deep cerebellar nuclei Adaptive threshold router 2025 Frontiers
Cortico-cerebellar loop Task consolidation pipeline Boven et al. 2024 Nature Comms

Full scientific audit with honest limitations: docs/architecture.md §15.

Quick start

# Clone & install
git clone https://github.com/QEout/digital-cerebellum.git
cd digital-cerebellum
pip install -e .

# Configure (fill in your API key)
cp config.yaml config.local.yaml
# Edit config.local.yaml → llm.api_key
from digital_cerebellum import DigitalCerebellum, CerebellumConfig
from digital_cerebellum.microzones.tool_call import ToolCallMicrozone

cb = DigitalCerebellum(CerebellumConfig.from_yaml())
cb.register_microzone(ToolCallMicrozone())

# Generic API — works with any registered microzone
result = cb.evaluate("tool_call", {
    "tool_name": "send_email",
    "tool_params": {"to": "alice", "body": "hello"},
})
print(result)  # {"safe": True, "confidence": 0.98, ...}

Supports any OpenAI-compatible LLM: Qwen, GPT, Claude, Ollama, etc.

Phase 0 validation results (210 tool calls, real Qwen 3.5 Flash)

Metric Target Actual
Fast-path ratio > 30% 96% PASS
Fast-path accuracy (blind test) > 60% CB-LLM agreement 100% (30/30) PASS
Fast-path latency < 100ms 50ms avg PASS
Speedup vs LLM > 5x 33.4x PASS
LLM cost savings 46% fewer API calls

Run the experiment yourself: python -m experiments.closed_loop

Project status

Phase 0 — Core engine + validation (complete)

  • Pattern separator (RFF with top-k sparsification)
  • K-head prediction engine (population coding → emergent confidence)
  • Dendritic masking per head (different feature subsets → genuine diversity)
  • 3-channel error comparator (SPE implemented; TPE/RPE interfaces ready)
  • Online learner (SGD + EWC + replay buffer)
  • Adaptive decision router (RPE-driven threshold)
  • Fluid memory v0 (decay + reconsolidation)
  • Cortex interface (OpenAI-compatible LLM)
  • Pluggable microzone architecture (universal cerebellar transform)
  • Tool-call safety microzone (first plugin)
  • Main pipeline (DigitalCerebellum class)
  • 23 unit tests passing
  • 210-step closed-loop validation with real LLM (4/4 metrics passed)
  • pip install -e . SDK packaging

See roadmap for Phase 1-3.

Architecture

Two documents cover everything:

  • docs/architecture.md — Biological mappings, system design, neuroscience corrections (v2), digital life panorama, application scenarios, scientific audit, competitive landscape, honest boundaries
  • docs/implementation.md — Component-level technical details, algorithms, dependencies, file structure

Key design decisions

Population coding, not sigmoid confidence. K=4 independent prediction heads. Confidence emerges from agreement — when heads disagree, uncertainty is real, not a learned artifact.

Three error channels, not one. Sensory prediction error updates the predictor. Temporal error updates the rhythm system. Reward error updates the router. Each drives different learning.

Continuous prediction, not classification. The engine predicts outcomes (continuous vectors), not labels. The safe/unsafe decision derives from continuous confidence. This satisfies the cerebellum's continuity constraint (Tsay & Ivry 2025).

EWC is an engineering approximation. Biological cerebellum prevents catastrophic forgetting via systems consolidation (cerebellum→cortex transfer), not weight regularization. Our task consolidation pipeline (Phase 1) is the more faithful mechanism. We're honest about this.

Run tests

pytest tests/ -v

License

MIT

Citation

If you use this work in research:

@software{digital_cerebellum_2026,
  title={Digital Cerebellum: A Cerebellar-Inspired Prediction-Correction Engine for LLM Agents},
  year={2026},
  url={https://github.com/QEout/digital-cerebellum}
}

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

digital_cerebellum-0.1.0.tar.gz (32.4 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

digital_cerebellum-0.1.0-py3-none-any.whl (33.1 kB view details)

Uploaded Python 3

File details

Details for the file digital_cerebellum-0.1.0.tar.gz.

File metadata

  • Download URL: digital_cerebellum-0.1.0.tar.gz
  • Upload date:
  • Size: 32.4 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.13.2

File hashes

Hashes for digital_cerebellum-0.1.0.tar.gz
Algorithm Hash digest
SHA256 5043f2c0026efc20bfaa438938d3ede16082a6fe85eafcb4f2416b621ef62632
MD5 8e99b4299873299bb902eaaa4c683cb1
BLAKE2b-256 3ec7ffeb387ddfa3f1d8048bca87ba75afda18c11c7a1a27ffe22e1f1da1afad

See more details on using hashes here.

File details

Details for the file digital_cerebellum-0.1.0-py3-none-any.whl.

File metadata

File hashes

Hashes for digital_cerebellum-0.1.0-py3-none-any.whl
Algorithm Hash digest
SHA256 7e7246bd048b46194c786268078a0ccccbef774a9f01346a57d2fea2e429e10f
MD5 d8aaa93ceebb79edd9b77698664342eb
BLAKE2b-256 6b35cbce42784df10b26f58af830eadaf8931a0d01ac5cadcacfcac5599b03f7

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page