fly-xox-malecns / webgpu /manifest.json
n4ze3m's picture
Add measured-graph WebGPU export and validation
2ca04b7 verified
Raw
History Blame
9.23 kB
{
"format": "fly-xox-webgpu-v1",
"neurons": 166700,
"edges": 25582938,
"steps": 4,
"readout_count": 2048,
"activity_samples": 128,
"arrays": {
"rowptr": {
"bytes": 666804,
"dtype": "uint32",
"shape": [
166701
],
"parts": [
{
"file": "rowptr.000.bin",
"bytes": 666804,
"sha256": "649f7b23e1cdbca64a500f9f364ebbb0da5889c8be75a9086e1d5ce29d888375"
}
]
},
"column": {
"bytes": 102331752,
"dtype": "uint32",
"shape": [
25582938
],
"parts": [
{
"file": "column.000.bin",
"bytes": 16777216,
"sha256": "db62672e4885d0bfae681b38070cae1c6b0c3205773865498ebe25e7f4459cbc"
},
{
"file": "column.001.bin",
"bytes": 16777216,
"sha256": "feb154bd9de73cc8b8f9b80674b8a02d963522b56136e14d31a693301e2cee30"
},
{
"file": "column.002.bin",
"bytes": 16777216,
"sha256": "ff7a62230404ca98f986d8999856389742df002a3eea30f83c4bfbec3a78ee71"
},
{
"file": "column.003.bin",
"bytes": 16777216,
"sha256": "7a87dd746121b525c88e610ba7ccac8d9db2ffba3b43fb006e07c8bbc10a6325"
},
{
"file": "column.004.bin",
"bytes": 16777216,
"sha256": "fff35cc2099c0fcddba4efa72d13da87000085a5376409d60fb8099d485bb3c5"
},
{
"file": "column.005.bin",
"bytes": 16777216,
"sha256": "1644e968f06686b8d7e4e4aceefdc1526df19c233c66f3545eb090cbc16eb824"
},
{
"file": "column.006.bin",
"bytes": 1668456,
"sha256": "16f0075043dedbc80f15691c3ae200d0f4f8c0f450a48eb3ff388b147e4a53b4"
}
]
},
"weight": {
"bytes": 102331752,
"dtype": "float32",
"shape": [
25582938
],
"parts": [
{
"file": "weight.000.bin",
"bytes": 16777216,
"sha256": "d2bc0f4a50cccadcf1327feff108cd3ff4bab96bf91b9a2ccca3880cc4831d03"
},
{
"file": "weight.001.bin",
"bytes": 16777216,
"sha256": "a9baca9748e90a1f800a59e494b4665f857c0319d2c4f0504bbfb7c5d220df66"
},
{
"file": "weight.002.bin",
"bytes": 16777216,
"sha256": "2bdf9f17022a6ef05707ef5d75d15a1e88995567a6cd1a6eecc5cb78c4d185ae"
},
{
"file": "weight.003.bin",
"bytes": 16777216,
"sha256": "6a2c629bdab941b738217ce6ca2cf07c37d1063d1eb18a7bf11743c4777a4599"
},
{
"file": "weight.004.bin",
"bytes": 16777216,
"sha256": "6d8478942597bab185f11159b33ca67336636818be0f2b1064d2262fb9bda729"
},
{
"file": "weight.005.bin",
"bytes": 16777216,
"sha256": "f62c5ee13e5918f885ec219e51c4316f0f3ad81942a7680365519bfe5e46d918"
},
{
"file": "weight.006.bin",
"bytes": 1668456,
"sha256": "380a0c88b101fdfbe5f5603a25bc6092965e72ca368b35243c71e05f8ffe07be"
}
]
},
"encoder": {
"bytes": 18003600,
"dtype": "float32",
"shape": [
166700,
27
],
"parts": [
{
"file": "encoder.000.bin",
"bytes": 16777216,
"sha256": "637308f4fe82841228c00d54bddcc103d8e4053100c2fcfb0ed7927c1191e7be"
},
{
"file": "encoder.001.bin",
"bytes": 1226384,
"sha256": "c8294c0565316a9417e7e881e0372baccc70e41253c90cf827181167722d29dd"
}
]
},
"parameters": {
"bytes": 1333600,
"dtype": "float32",
"shape": [
166700,
2
],
"parts": [
{
"file": "parameters.000.bin",
"bytes": 1333600,
"sha256": "f6fa3c11eb2b56f85ef2300f6a1ec39153017dc3aa0288421a176db66903d4d5"
}
]
},
"neurons": {
"bytes": 8192,
"dtype": "uint32",
"shape": [
2048
],
"parts": [
{
"file": "neurons.000.bin",
"bytes": 8192,
"sha256": "6f8a3950ff6a944e78313c775079e22b7e6b8fa9fc0359a53842400654b0d07b"
}
]
},
"projection": {
"bytes": 73728,
"dtype": "float32",
"shape": [
9,
2048
],
"parts": [
{
"file": "projection.000.bin",
"bytes": 73728,
"sha256": "cddc831cdf62d1c1db5cc93309ef5d5fc366ad8cce4166eaa51d6e5d05796f11"
}
]
}
},
"total_bytes": 224749428,
"source": {
"dataset": "MaleCNS v1.0",
"source": "https://male-cns.janelia.org/download/",
"license": "CC BY 4.0",
"sources": {
"annotations": {
"url": "https://storage.googleapis.com/flyem-male-cns/v1.0/connectome-data/flat-connectome/body-annotations-male-cns-v1.0-minconf-0.5.feather",
"bytes": 14483314,
"sha256": "2177e246113e4cfbf1e7772ec37c6da1955ff22e8063d0b1f833101f99a9a3b2"
},
"neurotransmitters": {
"url": "https://storage.googleapis.com/flyem-male-cns/v1.0/connectome-data/flat-connectome/body-neurotransmitters-male-cns-v1.0.feather",
"bytes": 43282834,
"sha256": "95c9289220663abeb3409f3ad9e5a7f8a53f8093f5139d15502cd08da8879621"
},
"edges": {
"url": "https://storage.googleapis.com/flyem-male-cns/v1.0/connectome-data/flat-connectome/connectome-weights-male-cns-v1.0-minconf-0.5.feather",
"bytes": 1051241946,
"sha256": "e35da783d1c686b2b58b3b87cd6a403ae43bfcfba8bff28e08ef752c1a56afc1"
}
},
"neurons": 166700,
"edges": 25582938,
"retained_synaptic_contacts": 124177617,
"raw_edge_rows": 151856684,
"raw_contacts": 311833243,
"sensory_neurons": 17937,
"neurotransmitters": {
"acetylcholine": 103720,
"glutamate": 29302,
"gaba": 22069,
"histamine": 7891,
"unclear": 2999,
"dopamine": 392,
"unknown": 178,
"octopamine": 101,
"serotonin": 48
},
"retention": "Assigned nonempty neuronal superclass; exclude explicit Glia. All released connections between retained IDs; no extra threshold.",
"sign_assumption": "GABA and glutamate negative; all other or unknown transmitters positive. Receptor-specific effects and neuromodulation unmodeled.",
"input": "Fixed random projection of 27 symbolic board channels into annotated sensory neurons. No learned input adapter; not optical vision.",
"biological_dynamics_validated": false
},
"results": {
"validation_selected_step": 1100,
"elapsed_seconds": 42.79798544710502,
"training": {
"positions": 3560,
"optimal_move_rate": 0.9817415475845337,
"illegal_moves": 0
},
"validation": {
"positions": 492,
"optimal_move_rate": 0.9004064798355103,
"illegal_moves": 0
},
"test": {
"positions": 468,
"optimal_move_rate": 0.9166666865348816,
"illegal_moves": 0
},
"trainable_parameters": 500100,
"changed_internal_parameters": {
"gain": 153090,
"bias": 165455,
"leak": 165455
},
"fixed_interfaces_unchanged": true,
"one_training_seed": true,
"split": "D4 symmetry groups of player-relative boards, SHA256 bucket; 80/10/10 approximate",
"device": "NVIDIA GeForce RTX 3090",
"peak_gpu_gb": 1.013288448,
"uniform_random_optimal_rate": 0.5691086691086691,
"untrained_test": {
"positions": 468,
"optimal_move_rate": 0.6282051205635071,
"illegal_moves": 0
},
"games": {
"random_fly_first": {
"wins": 474,
"draws": 26,
"losses": 0
},
"random_fly_second": {
"wins": 425,
"draws": 70,
"losses": 5
},
"minimax_fly_first": {
"wins": 0,
"draws": 500,
"losses": 0
},
"minimax_fly_second": {
"wins": 0,
"draws": 480,
"losses": 20
}
},
"games_note": "500 games per opponent and side, deterministic learned policy, seeded random opponent/tie choices. Complete games can include training boards; held-out position accuracy is reported separately."
},
"dynamics": "Four recurrent softsign rate updates by default; reset each move; no biological timing claim.",
"learned": "Internal per-neuron incoming gain, bias, and leak. Fixed input and output adapters.",
"webgpu_verified": true,
"webgpu_verification": {
"adapter": {
"vendor": "llvmpipe",
"architecture": "",
"device": "llvmpipe (LLVM 15.0.7, 256 bits)",
"description": "Mesa 23.2.1-1ubuntu3.1~22.04.4 (LLVM 15.0.7)",
"stackgroup_min_size": 4,
"stackgroup_max_size": 128,
"vendor_id": 65541,
"device_id": 0,
"adapter_type": "CPU",
"backend_type": "Vulkan"
},
"positions": 12,
"maximum_absolute_logit_error": 5.0067901611328125e-06,
"all_moves_match": true,
"median_move_ms": 762.1890138834715,
"cold_move_ms": 1157.2410590015352,
"backend": "wgpu-native executing the browser WGSL; not a browser end-to-end test",
"passed": true
}
}