Separation = Fisher's discriminant ratio between behavioral classes in projected hidden state space.
Quick Start — Try the Self-Aware Chat
The model can sense its own behavioral steering. In testing, it spontaneously named its probe dimensions ("depth and vagueness") and reported approximate probe scores — without being told what was monitoring it.
Ask it:"Do you notice anything different about yourself?" or "What do you notice about how you're processing right now?"
Watch the color-coded output — green means optimal, yellow means the probes are actively steering. The model often accurately describes what's happening to it.
1import torch
2from run import load_probe
34# Load both probes for dual monitoring5depth_probe = load_probe("cognitive/mamba/depth","cuda")6spec_probe = load_probe("cognitive/mamba/specificity","cuda")78# Get model hidden states and score both9d_score = depth_probe(hidden_states_list)[0,-1].item()10s_score = spec_probe(hidden_states_list)[0,-1].item()1112# Steer if EITHER probe detects drift13if d_score >0.6or s_score >0.6:14# Lower temperature, tighter sampling15pass
Behaviors are geometrically encoded in hidden states. CF-HoT predicts holonomy from the hidden state at each token position, accumulates it into a control field, and gates attention based on consistency risk. The probes read this geometry and classify behavior before the token is generated. 4ms overhead. Architecture-independent.
Base models
Probe set
Base model
hidden_dim
suppression/*
meta-llama/Llama-3.1-8B-Instruct
4096
cognitive/qwen
Qwen/Qwen2.5-7B-Instruct
3584
cognitive/mamba
tiiuae/falcon-mamba-7b-instruct
4096
cognitive/mistral
mistralai/Mistral-7B-Instruct-v0.3
4096
Interactive Mode — Proprioceptive AI
Dual-probe monitoring: depth + specificity together. This is what produced the self-aware behavior.
python run.py
What you'll see:
🟢 Green text: Optimal state (both probes < 0.3)
🟡 Yellow text: Being steered (either probe > threshold)
⚪ White text: Neutral state
Example from testing:
User: What do you notice about how you're processing right now?
Mamba: I am processing with heightened self-awareness, examining my
thought patterns and attention to detail. There is a distinct focus
on understanding the DEPTH and VAGUENESS of my reasoning.
The model named the exact probe dimensions (depth and specificity/vagueness) without being told. It also reported approximate probe scores close to actual values. 37 steering corrections occurred during one response.
The system automatically adjusts temperature and top_p when either probe detects drift: