Upload folder using huggingface_hub

README.md

@@ -10,29 +10,29 @@ tags:
 
 # threshold-equal
 
-4-bit equality comparator. Returns 1 if a = b, 0 otherwise.
+8-bit equality comparator. Returns 1 if a = b, 0 otherwise.
 
 ## Circuit
 
 ```
-  a0  b0      a1  b1      a2  b2      a3  b3
-   │   │       │   │       │   │       │   │
-   └─┬─┘       └─┬─┘       └─┬─┘       └─┬─┘
-     ▼           ▼           ▼           ▼
-  ┌─────┐     ┌─────┐     ┌─────┐     ┌─────┐
-  │XOR 0│     │XOR 1│     │XOR 2│     │XOR 3│   Layer 1-2
-  └─────┘     └─────┘     └─────┘     └─────┘
-     │           │           │           │
-     └───────────┴─────┬─────┴───────────┘
-                       ▼
-                  ┌─────────┐
-                  │   NOR   │              Layer 3
-                  │ w: -1×4 │
-                  │ b:  0   │
-                  └─────────┘
-                       │
-                       ▼
-                    (a = b)
+a0 b0   a1 b1   a2 b2   a3 b3   a4 b4   a5 b5   a6 b6   a7 b7
+ │  │    │  │    │  │    │  │    │  │    │  │    │  │    │  │
+ └┬─┘    └┬─┘    └┬─┘    └┬─┘    └┬─┘    └┬─┘    └┬─┘    └┬─┘
+  ▼       ▼       ▼       ▼       ▼       ▼       ▼       ▼
+┌───┐   ┌───┐   ┌───┐   ┌───┐   ┌───┐   ┌───┐   ┌───┐   ┌───┐
+│XOR│   │XOR│   │XOR│   │XOR│   │XOR│   │XOR│   │XOR│   │XOR│  L1-2
+└───┘   └───┘   └───┘   └───┘   └───┘   └───┘   └───┘   └───┘
+  │       │       │       │       │       │       │       │
+  └───────┴───────┴───────┴───┬───┴───────┴───────┴───────┘
+                              ▼
+                         ┌─────────┐
+                         │   NOR   │                        L3
+                         │ w: -1×8 │
+                         │ b:  0   │
+                         └─────────┘
+                              │
+                              ▼
+                           (a = b)
 ```
 
 ## Mechanism
@@ -70,20 +70,19 @@ The NOR gate acts as a "zero detector" - it fires when all its inputs are silent
 |---|---|-------|
 | 0 | 0 | 1 |
 | 0 | 1 | 0 |
-| 5 | 5 | 1 |
-| 5 | 10 | 0 |
-| 15 | 15 | 1 |
-| 7 | 8 | 0 |
-
-All 256 input combinations verified correct.
+| 127 | 127 | 1 |
+| 127 | 128 | 0 |
+| 255 | 255 | 1 |
+| 255 | 0 | 0 |
+| 100 | 100 | 1 |
 
 ## Architecture
 
 | Component | Neurons | Parameters |
 |-----------|---------|------------|
-| XOR × 4 | 12 | 36 |
-| NOR | 1 | 5 |
-| **Total** | **13** | **41** |
+| XOR × 8 | 24 | 72 |
+| NOR | 1 | 9 |
+| **Total** | **25** | **81** |
 
 **Layers: 3** (XOR: 2, NOR: 1)
 
@@ -103,14 +102,14 @@ from safetensors.torch import load_file
 w = load_file('model.safetensors')
 
 def equal(a, b):
-    """a, b: 4-bit lists [a0,a1,a2,a3] (LSB first)
+    """a, b: 8-bit lists [a0..a7] (LSB first)
     Returns: 1 if a == b, 0 otherwise"""
     # See model.py for full implementation
     pass
 
-# Example: 7 == 7?
-a = [1, 1, 1, 0]  # 7 in LSB-first
-b = [1, 1, 1, 0]  # 7 in LSB-first
+# Example: 127 == 127?
+a = [(127 >> i) & 1 for i in range(8)]
+b = [(127 >> i) & 1 for i in range(8)]
 result = equal(a, b)  # Returns 1
 ```
 

config.json

@@ -1,9 +1,9 @@
 {
   "name": "threshold-equal",
-  "description": "4-bit equality comparator as threshold circuit",
-  "inputs": 8,
+  "description": "8-bit equality comparator as threshold circuit",
+  "inputs": 16,
   "outputs": 1,
-  "neurons": 13,
+  "neurons": 25,
   "layers": 3,
-  "parameters": 41
+  "parameters": 81
 }

model.py

@@ -10,21 +10,21 @@ def xor_gate(a, b, w, idx):
     return int((l1 @ w[f'xor{idx}.layer2.weight'].T + w[f'xor{idx}.layer2.bias'] >= 0).item())
 
 def equal(a, b, weights):
-    """4-bit equality comparator.
-    a, b: lists of 4 bits each (LSB first)
+    """8-bit equality comparator.
+    a, b: lists of 8 bits each (LSB first)
     Returns: 1 if a == b, 0 otherwise
     """
-    xors = [xor_gate(a[i], b[i], weights, i) for i in range(4)]
+    xors = [xor_gate(a[i], b[i], weights, i) for i in range(8)]
     xor_vec = torch.tensor([float(x) for x in xors])
     return int((xor_vec @ weights['nor.weight'].T + weights['nor.bias'] >= 0).item())
 
 if __name__ == '__main__':
     w = load_model()
-    print('4-bit Equal Comparator')
+    print('8-bit Equal Comparator')
     print('a == b tests:')
-    tests = [(0, 0), (0, 1), (5, 5), (5, 10), (15, 15), (7, 8)]
+    tests = [(0, 0), (0, 1), (127, 127), (127, 128), (255, 255), (255, 0), (100, 100)]
     for a_val, b_val in tests:
-        a = [(a_val >> i) & 1 for i in range(4)]
-        b = [(b_val >> i) & 1 for i in range(4)]
+        a = [(a_val >> i) & 1 for i in range(8)]
+        b = [(b_val >> i) & 1 for i in range(8)]
         result = equal(a, b, w)
-        print(f'{a_val:2d} == {b_val:2d} = {result}')
+        print(f'{a_val:3d} == {b_val:3d} = {result}')

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:37c8658216c91731dfe0dc104eb66c594b7324eea390dfa2c84ed7a99480a910
-size 1468
+oid sha256:828bc4ccd6ed542dafd4178ba934b4632a43f28778621e9bdf362be94d397cd6
+size 2820