You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: benchmarks/README.md
+12-7Lines changed: 12 additions & 7 deletions
Original file line number
Diff line number
Diff line change
@@ -1,6 +1,6 @@
1
1
## Benchmarks
2
2
We validate the benchmark results provided in [HippoRAG](https://arxiv.org/abs/2405.14831), as well as comparing with other methods:
3
-
- NaiveRAG (vector dbs) using the embedder `text-embedding-3-small`
3
+
- NaiveRAG (vector dbs) using the OpenAI embedder `text-embedding-3-small`
4
4
-[LightRAG](https://github.com/HKUDS/LightRAG)
5
5
-[GraphRAG](https://github.com/gusye1234/nano-graphrag) (we use the implementation provided by `nano-graphrag`, based on the original [Microsoft GraphRAG](https://github.com/microsoft/graphrag))
6
6
@@ -12,12 +12,14 @@ We validate the benchmark results provided in [HippoRAG](https://arxiv.org/abs/2
12
12
|| VectorDB| 0.49| 0.32|
13
13
|| LightRAG| 0.47| 0.32|
14
14
|| GraphRAG| 0.75| 0.68|
15
-
||**Circlemind**| 0.96| 0.95|
15
+
||**Circlemind**|**0.96**|**0.95**|
16
16
| 101||||
17
17
|| VectorDB| 0.42| 0.23|
18
18
|| LightRAG| 0.45| 0.28|
19
19
|| GraphRAG| 0.73| 0.64|
20
-
||**Circlemind**| 0.93| 0.90|
20
+
||**Circlemind**|**0.93**|**0.90**|
21
+
22
+
**Circlemind is up to 4x more accurate than VectorDB RAG.**
21
23
22
24
**HotpotQA**
23
25
| # Queries | Method | All queries % |
@@ -26,27 +28,30 @@ We validate the benchmark results provided in [HippoRAG](https://arxiv.org/abs/2
26
28
|| VectorDB| 0.78|
27
29
|| LightRAG| 0.55|
28
30
|| GraphRAG| -*|
29
-
||**Circlemind**| 0.84|
31
+
||**Circlemind**|**0.84**|
30
32
31
33
*: crashes after half an hour of processing
32
34
33
-
We also briefly report the insertion times for the 2wikimultihopqa benchmark (~800 chunks):
35
+
Below, find the insertion times for the 2wikimultihopqa benchmark (~800 chunks):
34
36
| Method | Time (minutes) |
35
37
|:--------:|-----------------:|
36
38
| VectorDB|~0.3|
37
39
| LightRAG|~25|
38
40
| GraphRAG|~40|
39
41
|**Circlemind**|~1.5|
40
42
43
+
**Circlemind is 27x faster than GraphRAG while also being over 40% more accurate in retrieval.**
44
+
41
45
### Run it yourself
42
46
The scripts in this directory will generate and evaluate the 2wikimultihopqa datasets on a subsets of 51 and 101 queries with the same methodology as in the HippoRAG paper. In particular, we evaluate the retrieval capabilities of each method, mesauring the percentage of queries for which all the required evidence was retrieved. We preloaded the results so it is enough to run `evaluate_dbs.xx` to get the numbers. You can also run `create_dbs.xx` to regenerate the databases for the different methods.
47
+
43
48
A couple of NOTES:
44
49
- you will need to set an OPENAI_API_KEY;
45
-
- LightRAG and GraphRAG could take a while (~1 hour) to process;
50
+
- LightRAG and GraphRAG could take a over an 1 hour to process and they can be expensive;
46
51
- when pip installing LightRAG, not all dependencies are added; to run it we simply deleted all the imports of each missing dependency (since we use OpenAI they are not necessary).
47
52
- we also benchmarked on the HotpotQA dataset (we will soon release the code for that as well).
48
53
49
-
The output should looks similar to the following (the exact numbers could vary based on your graph configuration)
54
+
The output will look similar to the following (the exact numbers could vary based on your graph configuration)
50
55
```
51
56
Evaluation of the performance of different RAG methods on 2wikimultihopqa (51 queries)
0 commit comments