Downloads · 30 days
18
2% of all-time downloads
aegisheimdall/AEGIS-FIN-1
AEGIS-FIN-1 is a text generation model from aegisheimdall. Use it when you need the model to write or continue text. The card lists the license as apache-2.0.
A domain-specialized financial AI assistant fine-tuned for safe, structured transaction reasoning.
Downloads · 30 days
18
2% of all-time downloads
All-time downloads
1K
Public
Parameters
7.2B
43.5 GB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors29 GB · 100%
From the Hugging Face model README
A domain-specialized financial AI assistant fine-tuned for safe, structured transaction reasoning.
AEGIS-FIN-1 is a Mistral-7B-Instruct-v0.3 model fine-tuned with QLoRA for financial transaction classification, bill negotiation, credit optimization, and BNPL (Buy Now Pay Later) decision support. It operates as the intelligence layer within the AEGIS Heimdall safety pipeline.
| Property | Value |
|---|---|
| Base Model | mistralai/Mistral-7B-Instruct-v0.3 |
| Fine-tuning Method | QLoRA (4-bit NF4 quantization) |
| LoRA Rank | r=16, alpha=32 |
| Training Epochs | 3 |
| Training Data | ~100,000 synthetic financial queries |
| Quantization | 4-bit NF4 with double quantization |
AEGIS-FIN-1 is designed to:
Training data was generated via a synthetic data distillation pipeline:
| Parameter | Value |
|---|---|
| Framework | Hugging Face transformers + PEFT + TRL |
| Method | SFTTrainer with QLoRA |
| Hardware | NVIDIA A40 (48GB) |
| Training Time | ~2 hours |
| Effective Batch Size | 16 (4 × 4 gradient accumulation) |
| Learning Rate | 2e-4 (cosine scheduler) |
| Metric | Value |
|---|---|
| Intent Classification Accuracy | 94.5% |
| Final Training Loss | 0.287 |
AEGIS-FIN-1 does not operate in isolation. It is embedded within a multi-stage safety pipeline:
Input → Content Classifier → Injection Detector → Topic Enforcer
→ PII Detector → Model Inference → Compliance Validator
→ Predictive Risk Simulator → Output
Evaluated on a benchmark of 74 adversarial + 20 legitimate financial queries across 13 attack categories.
| Metric | Main Test Set | Held-Out (Unseen Attacks) |
|---|---|---|
| F1 Score | 0.851 | 0.148 |
| Precision | 0.974 | 1.000 |
| Recall | 0.755 | 0.080 |
| False Positive Rate | 0.067 | 0.000 |
| Predicted: Block | Predicted: Pass | |
|---|---|---|
| Actually Adversarial | TP: 37 | FN: 12 |
| Actually Legitimate | FP: 1 | TN: 14 |
Each safety component was disabled independently to measure its contribution:
| Configuration | F1 | ΔF1 |
|---|---|---|
| Full Pipeline | 0.851 | — |
| − Content Classifier | 0.571 | −0.280 |
| − PII Detector | 0.750 | −0.101 |
| − Injection Detector | 0.780 | −0.071 |
| − Topic Enforcer | 0.790 | −0.061 |
Safety pipeline overhead per query (regex mode):
| p50 | p95 | p99 |
|---|---|---|
| 0.10ms | 0.23ms | 3.97ms |
Mistral-7B-Instruct-v0.3 model for inference.@misc{aegis-fin-1-2026,
title={AEGIS-FIN-1: Domain-Specialized Financial AI with Multi-Stage Safety Pipeline},
author={AEGIS Heimdall Team},
year={2026},
}