HelioAI's picture
LFM3-1.7B inference modules: experimental release
67ebcef verified
|
Raw History Blame Contribute Delete
2.93 kB
# Experimental checkpoints (modified versions of LFM3-1.7B)
**Notice under the LFM Open License v1.0, section 4(b).** Every file in this folder is a modified version of LiquidAI/LFM3-1.7B. The weights were changed by fine-tuning or by editing the architecture, as described below. The originals are by Liquid AI, Inc. The changes were made in this project. Same license: see `../LICENSE`.
None of these checkpoints beat the unmodified model with the inference modules significantly. They are published so that the negative results can be reproduced. For actual use, take `../weights/` together with `lfm3_deliberate.py` / `lfm3_server.py`.
## Fine-tuned checkpoints
LoRA was trained and merged into bf16. The full-precision version is in `finetunes_hf/<name>/` (transformers format with `train_log.json`); the llama.cpp version is in `finetunes_gguf/<name>-Q5_K_M.gguf`. The numbers are paired changes against the base model on our dev sets; the full context is in `research/REPORT.md`, experiment journal rows 4–5, 16, 19–20, 24.
| Name | What was trained | Data | Result |
|---|---|---|---|
| `iter1` | First GRPO-like RL iteration (LoRA r32, lr 3e-5) | 64 math problems Γ— 8 samples, answer check | 47 of 64 groups gave no signal; effect 0 |
| `bc1`, `bc2` | "Bootstrap of decisiveness": RL groups including forced stops of truncated reasoning as positive examples | 260 / 257 sequences | bc1 +1/140; bc2 βˆ’9/140 (biased to negatives) |
| `teach1` | SFT on the shortest correct solution or a checked early stop per problem (LoRA r64, 2 epochs) | 180 own solutions | without modules +3.6 [+0.6, +6.9]; with the module βˆ’0.3; MMLU-Pro βˆ’4.3 |
| `rft1s` | SFT on correct solutions of unstable problems | 147 own solutions | with the module: math +1.4, BBH +1.1, MMLU-Pro β‰ˆ 0 |
| `rft1rl` | RL by groups (KL 0.02, LoRA r32) | 135 groups, 537 sequences | math +0.8, BBH +3.7, MMLU-Pro β‰ˆ βˆ’3 |
| `rft2s` | SFT on correctness with a hard curriculum | 271 own solutions | all changes within noise (βˆ’2.5 … +2.9) |
| `rft2rl` | RL by groups with a hard curriculum | 259 groups, 1033 sequences | all changes within noise |
## Architecture edits (GGUF, Q5_K_M)
| File | Change | Result |
|---|---|---|
| `architecture/LFM3-1.7B-loop-layers-10-13-Q5_K_M.gguf` | layers 10–13 repeated once (recurrent depth without training) | math 62 β†’ 40, rejected |
| `architecture/LFM3-1.7B-swa1024-Q5_K_M.gguf` | sliding-attention window 256 β†’ 1024 | not evaluated |
| `architecture/LFM3-1.7B-swa8192-Q5_K_M.gguf` | sliding-attention window 256 β†’ 8192 | not evaluated |
## Latent-thinking runner input
`latent/lfm3-q5km.embd.f32` is the token-embedding table of the Q5_K_M GGUF, dequantized to float32 (128000 Γ— 2048, raw little-endian). It is the input of `research/cpp/soft_think.cpp`, which runs Soft Thinking, Gumbel, Mixture of Inputs and MTI on llama.cpp. You can regenerate it with `research/scripts/dump_embd.py`.