Upload model, config, and documentation

README.md

@@ -0,0 +1,77 @@
+---
+tags:
+- pattern-classification
+- multi-label-classification
+datasets:
+- maximuspowers/muat-mean-std
+---
+
+# Pattern Classifier
+
+This model was trained to classify which patterns a subject model was trained on, based on neuron activation signatures.
+
+## Dataset
+
+- **Training Dataset**: [maximuspowers/muat-mean-std](https://huggingface.co/datasets/maximuspowers/muat-mean-std)
+- **Input Mode**: signature
+- **Number of Patterns**: 14
+
+## Patterns
+
+The model predicts which of the following 14 patterns the subject model was trained to classify as positive:
+
+1. `palindrome`
+2. `sorted_ascending`
+3. `sorted_descending`
+4. `alternating`
+5. `contains_abc`
+6. `starts_with`
+7. `ends_with`
+8. `no_repeats`
+9. `has_majority`
+10. `increasing_pairs`
+11. `decreasing_pairs`
+12. `vowel_consonant`
+13. `first_last_match`
+14. `mountain_pattern`
+
+## Model Architecture
+
+- **Signature Encoder**: [512, 256, 256, 128]
+- **Activation**: relu
+- **Dropout**: 0.2
+- **Batch Normalization**: True
+
+## Training Configuration
+
+- **Optimizer**: adam
+- **Learning Rate**: 0.001
+- **Batch Size**: 16
+- **Loss Function**: BCE with Logits (with pos_weight for training, unweighted for validation)
+
+## Test Set Performance
+
+- **F1 Macro**: 0.2832
+- **F1 Micro**: 0.2498
+- **Hamming Accuracy**: 0.7306
+- **Exact Match Accuracy**: 0.0140
+- **BCE Loss**: 0.4844
+
+### Per-Pattern Performance (Test Set)
+
+| Pattern | Precision | Recall | F1 Score |
+|---------|-----------|--------|----------|
+| palindrome | 11.1% | 89.8% | 19.8% |
+| sorted_ascending | 59.7% | 56.6% | 58.1% |
+| sorted_descending | 15.8% | 66.2% | 25.5% |
+| alternating | 19.8% | 72.4% | 31.1% |
+| contains_abc | 30.8% | 57.6% | 40.1% |
+| starts_with | 9.1% | 59.4% | 15.8% |
+| ends_with | 10.3% | 73.8% | 18.1% |
+| no_repeats | 17.8% | 32.1% | 22.9% |
+| has_majority | 33.3% | 60.5% | 43.0% |
+| increasing_pairs | 23.3% | 35.3% | 28.1% |
+| decreasing_pairs | 19.4% | 60.9% | 29.5% |
+| vowel_consonant | 9.8% | 76.9% | 17.4% |
+| first_last_match | 15.3% | 96.6% | 26.5% |
+| mountain_pattern | 15.3% | 31.7% | 20.6% |

best_model.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:64c68dc903e28b122d405966fd3d9c471a17b9da24d2d87b46959bbb797c5911
+size 4439480

config.yaml

@@ -0,0 +1,107 @@
+dataloader:
+  num_workers: 0
+  pin_memory: true
+dataset:
+  cache_dir: .cache/classifier_data
+  hf_dataset: maximuspowers/muat-mean-std
+  input_mode: signature
+  max_dimensions:
+    max_layers: 13
+    max_neurons_per_layer: 8
+    max_sequence_length: 5
+  neuron_profile:
+    methods:
+      mean: {}
+      std: {}
+  patterns:
+  - palindrome
+  - sorted_ascending
+  - sorted_descending
+  - alternating
+  - contains_abc
+  - starts_with
+  - ends_with
+  - no_repeats
+  - has_majority
+  - increasing_pairs
+  - decreasing_pairs
+  - vowel_consonant
+  - first_last_match
+  - mountain_pattern
+  random_seed: 42
+  test_split: 0.1
+  train_split: 0.8
+  val_split: 0.1
+device:
+  type: auto
+evaluation:
+  decision_threshold: 0.5
+  metrics:
+  - accuracy_exact_match
+  - accuracy_hamming
+  - precision_macro
+  - recall_macro
+  - f1_macro
+  - f1_micro
+  per_pattern_metrics: true
+hub:
+  enabled: true
+  private: false
+  push_frequency: epoch
+  push_logs: true
+  push_metrics: true
+  push_model: true
+  repo_id: maximuspowers/muat-mean-std-classifier
+  token: <REDACTED>
+logging:
+  checkpoint:
+    enabled: true
+    mode: max
+    monitor: val_f1_macro
+    save_best_only: true
+    save_dir: ./checkpoints/classifier_mean_std
+  tensorboard:
+    enabled: true
+    log_dir: ./runs/classifier_mean_std
+    log_interval: 10
+  verbose: true
+model:
+  fusion:
+    activation: relu
+    dropout: 0.2
+    hidden_dims:
+    - 128
+    - 64
+  output:
+    num_patterns: 14
+  signature_encoder:
+    activation: relu
+    dropout: 0.2
+    hidden_dims:
+    - 512
+    - 256
+    - 256
+    - 128
+  use_batch_norm: true
+  weight_encoder:
+    activation: relu
+    dropout: 0.2
+training:
+  batch_size: 16
+  early_stopping:
+    enabled: true
+    mode: min
+    monitor: val_loss
+    patience: 50
+  epochs: 1000
+  learning_rate: 0.001
+  loss: bce_with_logits
+  lr_scheduler:
+    enabled: true
+    factor: 0.5
+    min_lr: 1.0e-05
+    patience: 20
+    type: reduce_on_plateau
+  optimizer: adam
+  pos_weight: null
+  weight_decay: 0.0001