reubk commited on
Commit
fd7a290
·
verified ·
1 Parent(s): 08f18a2

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +26 -1
README.md CHANGED
@@ -10,4 +10,29 @@ tags:
10
  - agent
11
  - lora
12
  - finetune
13
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
10
  - agent
11
  - lora
12
  - finetune
13
+ ---
14
+
15
+
16
+ Testing a QLoRA adaptor for [allenai/MolmoE-1B-0924](https://huggingface.co/allenai/MolmoE-1B-0924),
17
+
18
+ Targets top 10 experts that are activated when pointing is involved and image pooling and projection layers of Vision backbone
19
+
20
+ Trained on 47 screenshots of a low-poly video game with ragdoll casualties
21
+
22
+ Evaluated on 44 screenshots of aforementioned video game
23
+
24
+ Molmo has an edge case where it declares there are no humans in an image:
25
+ ![img1 (2)](https://cdn-uploads.huggingface.co/production/uploads/6367f8dd46919b9619bc7bf2/8zsuqnz-QCTamBDOgWGM-.png)
26
+
27
+ This custom QLoRA successfully reduces the occurance of these cases
28
+ ![img1 (1)](https://cdn-uploads.huggingface.co/production/uploads/6367f8dd46919b9619bc7bf2/-HENqZx5SiLYX35tx3ADs.png)
29
+
30
+ However, pointing to non-human objects is observed to increase.
31
+
32
+ Comparison of Model performance with and without QLora on Eval dataset
33
+ |Model| MolmoE-1B | MolmoE-1B w/ QLora |
34
+ |----------|------|------|
35
+ | Precision | 82.4 | 81.5 |
36
+ | Recall | 63.5 | 72.1 |
37
+
38
+ Dataset: [reubk/RavenfieldDataset](https://huggingface.co/datasets/reubk/RavenfieldDataset)