From e51430020af365a92ab61c99f5152046fe67ef86 Mon Sep 17 00:00:00 2001 From: ModelHub XC Date: Tue, 21 Jul 2026 07:50:09 +0800 Subject: [PATCH] =?UTF-8?q?=E5=88=9D=E5=A7=8B=E5=8C=96=E9=A1=B9=E7=9B=AE?= =?UTF-8?q?=EF=BC=8C=E7=94=B1ModelHub=20XC=E7=A4=BE=E5=8C=BA=E6=8F=90?= =?UTF-8?q?=E4=BE=9B=E6=A8=A1=E5=9E=8B?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Model: Ayansk11/FinSenti-MobileLLM-R1-950M Source: Original Platform --- .gitattributes | 36 +++++++ README.md | 226 +++++++++++++++++++++++++++++++++++++++++ chat_template.jinja | 11 ++ config.json | 96 +++++++++++++++++ generation_config.json | 12 +++ model.safetensors | 3 + tokenizer.json | 3 + tokenizer_config.json | 21 ++++ 8 files changed, 408 insertions(+) create mode 100644 .gitattributes create mode 100644 README.md create mode 100644 chat_template.jinja create mode 100644 config.json create mode 100644 generation_config.json create mode 100644 model.safetensors create mode 100644 tokenizer.json create mode 100644 tokenizer_config.json diff --git a/.gitattributes b/.gitattributes new file mode 100644 index 0000000..52373fe --- /dev/null +++ b/.gitattributes @@ -0,0 +1,36 @@ +*.7z filter=lfs diff=lfs merge=lfs -text +*.arrow filter=lfs diff=lfs merge=lfs -text +*.bin filter=lfs diff=lfs merge=lfs -text +*.bz2 filter=lfs diff=lfs merge=lfs -text +*.ckpt filter=lfs diff=lfs merge=lfs -text +*.ftz filter=lfs diff=lfs merge=lfs -text +*.gz filter=lfs diff=lfs merge=lfs -text +*.h5 filter=lfs diff=lfs merge=lfs -text +*.joblib filter=lfs diff=lfs merge=lfs -text +*.lfs.* filter=lfs diff=lfs merge=lfs -text +*.mlmodel filter=lfs diff=lfs merge=lfs -text +*.model filter=lfs diff=lfs merge=lfs -text +*.msgpack filter=lfs diff=lfs merge=lfs -text +*.npy filter=lfs diff=lfs merge=lfs -text +*.npz filter=lfs diff=lfs merge=lfs -text +*.onnx filter=lfs diff=lfs merge=lfs -text +*.ot filter=lfs diff=lfs merge=lfs -text +*.parquet filter=lfs diff=lfs merge=lfs -text +*.pb filter=lfs diff=lfs merge=lfs -text +*.pickle filter=lfs diff=lfs merge=lfs -text +*.pkl filter=lfs diff=lfs merge=lfs -text +*.pt filter=lfs diff=lfs merge=lfs -text +*.pth filter=lfs diff=lfs merge=lfs -text +*.rar filter=lfs diff=lfs merge=lfs -text +*.safetensors filter=lfs diff=lfs merge=lfs -text +saved_model/**/* filter=lfs diff=lfs merge=lfs -text +*.tar.* filter=lfs diff=lfs merge=lfs -text +*.tar filter=lfs diff=lfs merge=lfs -text +*.tflite filter=lfs diff=lfs merge=lfs -text +*.tgz filter=lfs diff=lfs merge=lfs -text +*.wasm filter=lfs diff=lfs merge=lfs -text +*.xz filter=lfs diff=lfs merge=lfs -text +*.zip filter=lfs diff=lfs merge=lfs -text +*.zst filter=lfs diff=lfs merge=lfs -text +*tfevents* filter=lfs diff=lfs merge=lfs -text +tokenizer.json filter=lfs diff=lfs merge=lfs -text diff --git a/README.md b/README.md new file mode 100644 index 0000000..437aa67 --- /dev/null +++ b/README.md @@ -0,0 +1,226 @@ +--- +license: apache-2.0 +language: + - en +base_model: facebook/MobileLLM-R1-950M +datasets: + - Ayansk11/FinSenti-Dataset +pipeline_tag: text-generation +library_name: transformers +tags: + - finance + - financial-sentiment + - sentiment-analysis + - chain-of-thought + - reasoning + - grpo + - sft + - lora + - finsenti +--- +# FinSenti-MobileLLM-R1-950M + +FinSenti-MobileLLM-R1-950M is a 0.9B-parameter model fine-tuned to +read short financial text (headlines, earnings snippets, market commentary) +and explain its read of them before settling on positive, negative, or +neutral. It's Meta's purpose-built mobile model. The architecture is shaped for on-device inference (compact embeddings, untied lm_head, shared attention layers) and FinSenti's recipe lifts the financial-sentiment quality without changing that footprint. + +The model is part of the [FinSenti +collection](https://huggingface.co/collections/Ayansk11/finsenti), a +scaling study of small models trained on the same data with the same recipe. + +## What it's good at + +- Classifying short financial text (1-3 sentences) into positive / negative + / neutral +- Producing a short reasoning chain you can read or log +- Following a strict `......` output + format that's easy to parse downstream + +It was trained on news-style headlines and earnings snippets in English, so +that's where it shines. Outside that domain you'll see the format hold up +but the labels get noisier. + +## How it was trained + +Two-stage recipe, same across the whole FinSenti family: + +1. **SFT** on the SFT train slice from the [FinSenti + dataset](https://huggingface.co/datasets/Ayansk11/FinSenti-Dataset) + (~15.2K balanced training samples, drawn from a + 50.8K-sample pool with held-out val/test splits, chain-of-thought + targets generated by a teacher model and filtered for label agreement). + This stage took about 1.6 hours on a single A100 80GB + for this model. +2. **GRPO** with four reward functions (sentiment correctness, format + compliance, reasoning quality, output consistency), each weighted equally + for a maximum reward of 4.0. The training budget was 3000 + steps with early stopping; the best checkpoint landed near step + ~200 with a mean reward of approximately + **3.29 / 4.0** on the validation slice. + +Trainer stack: PEFT + bitsandbytes (no Unsloth - llama4_text arch unsupported), using Unsloth's pre-quantized mirror +[`facebook/MobileLLM-R1-950M`](https://huggingface.co/facebook/MobileLLM-R1-950M) as the +loading shortcut for the upstream +[`facebook/MobileLLM-R1-950M`](https://huggingface.co/facebook/MobileLLM-R1-950M) +weights. LoRA adapters (r=16, alpha=32) were +trained on the attention and MLP projection layers, then merged into the +base weights before export, so this repo is a self-contained model and +doesn't need PEFT to load. + +## Quick start + +Standard `transformers` usage: + +```python +from transformers import AutoModelForCausalLM, AutoTokenizer +import torch + +model_id = "Ayansk11/FinSenti-MobileLLM-R1-950M" +tok = AutoTokenizer.from_pretrained(model_id) +model = AutoModelForCausalLM.from_pretrained( + model_id, torch_dtype=torch.bfloat16, device_map="auto" +) + +system = ( + "You are a financial sentiment analyst. For each headline you receive, " + "write a short reasoning chain inside ... tags, " + "then give a single label inside ... tags. The label " + "must be exactly one of: positive, negative, neutral." +) +user = "Apple beats Q4 estimates as iPhone sales jump 12% year over year." + +messages = [ + {"role": "system", "content": system}, + {"role": "user", "content": user}, +] +prompt = tok.apply_chat_template(messages, tokenize=False, add_generation_prompt=True) + +inputs = tok(prompt, return_tensors="pt").to(model.device) +out = model.generate(**inputs, max_new_tokens=256, do_sample=False) +print(tok.decode(out[0][inputs.input_ids.shape[1]:], skip_special_tokens=True)) +``` + +Expected output (your reasoning text will vary; the label should match): + +``` + +Beating estimates is a positive earnings surprise. A 12% YoY iPhone sales jump in the company's biggest product line points to demand strength. Both signals push the read positive. + +positive +``` + +## Prompt format + +The model expects the system prompt above, verbatim is best. The user turn +is the headline or short snippet you want classified. Output is two XML-ish +blocks in this order: `...` then +`...`. The `` content is one of `positive`, +`negative`, or `neutral` (lowercase, no punctuation). + +If you want labels only and don't care about the reasoning, you can stop +generation as soon as you see `` to save tokens. + +## Performance notes + +The training reward (max 4.0) hit **3.29** on the +held-out validation slice. That breaks down across the four reward +functions roughly as: + +- Sentiment correctness: dominant contributor; the model gets the label + right on the validation split most of the time +- Format compliance: near-saturated by the end of GRPO; the model almost + always produces well-formed `` and `` tags +- Reasoning quality: judged on length and presence of finance-relevant + signal words; this one's the noisiest of the four +- Consistency: rewards stable labels across paraphrases of the same headline + +Numbers on standard finance benchmarks (FPB, FiQA, Twitter Financial News) +are forthcoming and will be added once the eval pipeline lands. + +## Hardware + +At bf16 the weights are about 1.8 GB on disk and need ~3 GB of GPU memory for batch=1 inference. CPU inference is fine too: on a modern laptop you'll get a few tokens per second with the bf16 weights, and 15-30 tok/s with the GGUF Q4_K_M build. + +## Limitations + +A few things this model isn't built for: + +- **Long documents.** Training context was capped at 2048 + tokens. Anything much longer than a few paragraphs is out of distribution. +- **Multi-asset reasoning.** It classifies the sentiment of a single piece + of text. It won't aggregate across multiple headlines or weigh sources. +- **Numerical reasoning.** It can read "beats by 12%" and call that + positive, but it isn't doing math. Don't ask it to forecast. +- **Languages other than English.** Training data was English only. +- **Background knowledge.** If the headline needs you to know what a + company does, the model only has whatever was in its base pretraining. + It can't look anything up. +- **Three labels, hard cutoffs.** The output space is positive / negative / + neutral. If you need a 5-class scale or a continuous score, you'll need + to retrain or post-process. + +## Training details + +| | | +|---|---| +| Upstream base model | [facebook/MobileLLM-R1-950M](https://huggingface.co/facebook/MobileLLM-R1-950M) | +| Loading mirror | [facebook/MobileLLM-R1-950M](https://huggingface.co/facebook/MobileLLM-R1-950M) (Unsloth's pre-quantized copy) | +| Dataset | [Ayansk11/FinSenti-Dataset](https://huggingface.co/datasets/Ayansk11/FinSenti-Dataset) (~15.2K train per stage, 50.8K total across splits) | +| SFT length | ~1.6 hours on A100 80GB | +| GRPO budget | 3000 steps with early stopping (best near step ~200) | +| Best GRPO reward | ~3.29 / 4.0 | +| Adapter | LoRA (r=16, alpha=32) on q/k/v/o/gate/up/down projections | +| Sequence length | 2048 | +| Optimizer | AdamW (8-bit), cosine LR schedule | +| Hardware | NVIDIA A100 80GB (Indiana University BigRed200 cluster) | +| Frameworks | PEFT + bitsandbytes (no Unsloth - llama4_text arch unsupported) | + +## Related FinSenti models + +Other sizes and bases trained with the same recipe: + +- **Qwen3**: [Qwen3-0.6B](https://huggingface.co/Ayansk11/FinSenti-Qwen3-0.6B), [Qwen3-1.7B](https://huggingface.co/Ayansk11/FinSenti-Qwen3-1.7B), [Qwen3-4B](https://huggingface.co/Ayansk11/FinSenti-Qwen3-4B), [Qwen3-8B](https://huggingface.co/Ayansk11/FinSenti-Qwen3-8B) +- **Qwen3.5**: [Qwen3.5-0.8B](https://huggingface.co/Ayansk11/FinSenti-Qwen3.5-0.8B), [Qwen3.5-2B](https://huggingface.co/Ayansk11/FinSenti-Qwen3.5-2B), [Qwen3.5-4B](https://huggingface.co/Ayansk11/FinSenti-Qwen3.5-4B), [Qwen3.5-9B](https://huggingface.co/Ayansk11/FinSenti-Qwen3.5-9B) +- **DeepSeek**: [DeepSeek-R1-1.5B](https://huggingface.co/Ayansk11/FinSenti-DeepSeek-R1-1.5B) +- **Tiny-LLM**: [Tiny-LLM-10M](https://huggingface.co/Ayansk11/FinSenti-Tiny-LLM-10M) +- **Llama-3**: [Llama-3.2-1B](https://huggingface.co/Ayansk11/FinSenti-Llama-3.2-1B) +- **SmolLM**: [SmolLM-1.7B](https://huggingface.co/Ayansk11/FinSenti-SmolLM-1.7B) + +There's a GGUF build of this same model at +[Ayansk11/FinSenti-MobileLLM-R1-950M-GGUF](https://huggingface.co/Ayansk11/FinSenti-MobileLLM-R1-950M-GGUF) for Ollama and +llama.cpp, and the dataset itself is at +[Ayansk11/FinSenti-Dataset](https://huggingface.co/datasets/Ayansk11/FinSenti-Dataset). + +If you're picking a size, a rough guide: + +- **Need it on a phone or browser?** Look at the smallest model in the + group (Qwen3-0.6B) or its GGUF. +- **Laptop with no GPU?** Any model up to ~2B as Q4_K_M GGUF works. +- **Single 8-12 GB GPU?** The 1.5B-4B sizes are the sweet spot. +- **Server or workstation?** The 8B / 9B variants give the best reasoning + but need the memory. + +## Citation + +If you use this model in research, please cite: + +```bibtex +@misc{shaikh2026finsenti, + title = {FinSenti: Small Language Models for Financial Sentiment with Chain-of-Thought Reasoning}, + author = {Shaikh, Ayan}, + year = {2026}, + url = {https://huggingface.co/collections/Ayansk11/finsenti}, + note = {Indiana University} +} +``` + +## License + +Apache 2.0, same as the base model. + +## Acknowledgements + +Trained on the Indiana University BigRed200 cluster. +Thanks to the Unsloth and TRL teams for the trainer stack, and to the +Qwen / DeepSeek teams for the base models. diff --git a/chat_template.jinja b/chat_template.jinja new file mode 100644 index 0000000..a9c2e0e --- /dev/null +++ b/chat_template.jinja @@ -0,0 +1,11 @@ +{% for message in messages %}{% if message['role'] == 'system' %}### System +{{ message['content'] }} + +{% elif message['role'] == 'user' %}### Input +{{ message['content'] }} + +{% elif message['role'] == 'assistant' %}### Response +{{ message['content'] }}{{ eos_token }} + +{% endif %}{% endfor %}{% if add_generation_prompt %}### Response +{% endif %} \ No newline at end of file diff --git a/config.json b/config.json new file mode 100644 index 0000000..a040a3e --- /dev/null +++ b/config.json @@ -0,0 +1,96 @@ +{ + "architectures": [ + "Llama4ForCausalLM" + ], + "attention_bias": false, + "attention_chunk_size": 32768, + "attention_dropout": 0.0, + "attn_scale": 0.1, + "attn_temperature_tuning": false, + "bos_token_id": 128000, + "dtype": "bfloat16", + "eos_token_id": [ + 128001, + 128008, + 128009 + ], + "floor_scale": 8192, + "for_llm_compressor": false, + "head_dim": 64, + "hidden_act": "silu", + "hidden_size": 1536, + "initializer_range": 0.02, + "interleave_moe_layer_step": 0, + "intermediate_size": 8192, + "intermediate_size_mlp": 6144, + "layer_types": [ + "full_attention", + "full_attention", + "full_attention", + "full_attention", + "full_attention", + "full_attention", + "full_attention", + "full_attention", + "full_attention", + "full_attention", + "full_attention", + "full_attention", + "full_attention", + "full_attention", + "full_attention", + "full_attention", + "full_attention", + "full_attention", + "full_attention", + "full_attention", + "full_attention", + "full_attention" + ], + "max_position_embeddings": 32768, + "model_type": "llama4_text", + "moe_layers": [], + "no_rope_layers": [ + 1, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + 1 + ], + "num_attention_heads": 24, + "num_experts_per_tok": 0, + "num_hidden_layers": 22, + "num_key_value_heads": 6, + "num_local_experts": 0, + "output_router_logits": false, + "pad_token_id": null, + "rms_norm_eps": 1e-05, + "rope_parameters": { + "rope_theta": 8000000.0, + "rope_type": "default" + }, + "router_aux_loss_coef": 0.001, + "router_jitter_noise": 0.0, + "tie_word_embeddings": false, + "transformers_version": "5.2.0", + "use_cache": true, + "use_qk_norm": true, + "vocab_size": 128257 +} diff --git a/generation_config.json b/generation_config.json new file mode 100644 index 0000000..a6a8e14 --- /dev/null +++ b/generation_config.json @@ -0,0 +1,12 @@ +{ + "_from_model_config": true, + "bos_token_id": 128000, + "cache_implementation": "dynamic", + "eos_token_id": [ + 128001, + 128008, + 128009 + ], + "tie_word_embeddings": false, + "transformers_version": "5.2.0" +} diff --git a/model.safetensors b/model.safetensors new file mode 100644 index 0000000..484cab5 --- /dev/null +++ b/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4bda94acb95e988dcfeaf69b751e9002125cf440ac3dbe71b7205426a9f6e892 +size 2293403880 diff --git a/tokenizer.json b/tokenizer.json new file mode 100644 index 0000000..870f2cb --- /dev/null +++ b/tokenizer.json @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6ebafa4822d64592396e4b0264db1bffc05d1765761bef03dfaeb670f93f5f05 +size 17210564 diff --git a/tokenizer_config.json b/tokenizer_config.json new file mode 100644 index 0000000..349d808 --- /dev/null +++ b/tokenizer_config.json @@ -0,0 +1,21 @@ +{ + "backend": "tokenizers", + "bos_token": "<|begin_of_text|>", + "clean_up_tokenization_spaces": true, + "eos_token": "<|eot_id|>", + "is_local": true, + "max_length": 512, + "model_input_names": [ + "input_ids", + "attention_mask" + ], + "model_max_length": 32768, + "pad_to_multiple_of": null, + "pad_token": "<|finsenti_pad|>", + "pad_token_type_id": 0, + "padding_side": "right", + "stride": 0, + "tokenizer_class": "TokenizersBackend", + "truncation_side": "right", + "truncation_strategy": "longest_first" +}