Create README.md
Browse files
README.md
ADDED
@@ -0,0 +1,50 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
license: apache-2.0
|
3 |
+
datasets:
|
4 |
+
- gbharti/finance-alpaca
|
5 |
+
language:
|
6 |
+
- en
|
7 |
+
library_name: transformers
|
8 |
+
tags:
|
9 |
+
- finance
|
10 |
+
widget:
|
11 |
+
- text: "Is this headline positive or negative? Headline: Australian Tycoon Forrest Shuts Nickel Mines After Prices Crash."
|
12 |
+
example_title: "Sentiment analysis"
|
13 |
+
- text: "Aluminum price per KG is 50$. Forecast max: +1$ min:+0.3$. What should be the current price of aluminum?"
|
14 |
+
example_title: "Forecast"
|
15 |
+
---
|
16 |
+
|
17 |
+
# Fin-RWKV: Attention Free Financal Expert (WIP)
|
18 |
+
Fin-RWKV is a cutting-edge, attention-free model designed specifically for financial analysis and prediction. Developed as part of a MindsDB Hackathon, this model leverages the simplicity and efficiency of the RWKV architecture to process financial data, providing insights and forecasts with remarkable accuracy. Fin-RWKV is tailored for professionals and enthusiasts in the finance sector who seek to integrate advanced deep learning techniques into their financial analyses.
|
19 |
+
|
20 |
+
## Use Cases
|
21 |
+
- Sentiment analysis
|
22 |
+
- Forecast
|
23 |
+
- Product Pricing
|
24 |
+
|
25 |
+
## Features
|
26 |
+
- Attention-Free Architecture: Utilizes the RWKV (Recurrent Weighted Kernel-based) model, which bypasses the complexity of attention mechanisms while maintaining high performance.
|
27 |
+
- Lower Costs: 10x to over a 100x+ lower inference cost, 2x to 10x lower training cost
|
28 |
+
- Tinyyyy: Lightweight enough to run on CPUs in real-time bypassing the GPU - and is able to run on your laptop today
|
29 |
+
- Finance-Specific Training: Trained on the gbharti/finance-alpaca dataset, ensuring that the model is finely tuned for financial data analysis.
|
30 |
+
- Transformers Library Integration: Built on the popular 'transformers' library, ensuring easy integration with existing ML pipelines and applications.
|
31 |
+
|
32 |
+
## Competing Against
|
33 |
+
| Name | Param Count | Cost | Inference Cost |
|
34 |
+
|---------------|-------------|------|----------------|
|
35 |
+
| Fin-RWKV | 1B5 | $3 | Free on HuggingFace 🤗 & Low-End CPU |
|
36 |
+
| [BloombergGPT](https://www.bloomberg.com/company/press/bloomberggpt-50-billion-parameter-llm-tuned-finance/) | 50 Billion | $1.3 million | Enterprise GPUs |
|
37 |
+
| [FinGPT](https://huggingface.co/FinGPT) | 7 Bilion | $302.4 | Consumer GPUs |
|
38 |
+
|
39 |
+
|
40 |
+
| Architecture | Status | Compute Efficiency | Largest Model | Trained Token | Link |
|
41 |
+
|--------------|--------|--------------------|---------------|---------------|------|
|
42 |
+
| (Fin)RWKV | In Production | O ( N ) | 14B | 500B++ (the pile+) | [Paper](https://arxiv.org/abs/2305.13048) |
|
43 |
+
| Ret Net (Microsoft) | Research | O ( N ) | 6.7B | 100B (mixed) | [Paper](https://arxiv.org/abs/2307.08621) |
|
44 |
+
| State Space (Stanford) | Prototype | O ( Log N ) | 355M | 15B (the pile, subset) | [Paper](https://arxiv.org/abs/2302.10866) |
|
45 |
+
| Liquid (MIT) | Research | - | <1M | - | [Paper](https://arxiv.org/abs/2302.10866) |
|
46 |
+
| Transformer Architecture (included for contrasting reference) | In Production | O ( N^2 ) | 800B (est) | 13T++ (est) | - |
|
47 |
+
|
48 |
+
<img src="https://cdn-uploads.huggingface.co/production/uploads/631ea4247beada30465fa606/7vAOYsXH1vhTyh22o6jYB.png" width="500" alt="Inference computational cost vs. Number of tokens">
|
49 |
+
|
50 |
+
_Note: Needs more data and training, testing purposes only._
|