Downloads · 30 days
0
karthifde/apexfi-fraud-detection
apexfi-fraud-detection is a machine learning model from karthifde. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for custom. The card lists the license as mit.
Two independent hybrid fraud-detection models — LightGBM (tabular) + GraphSAGE (graph neural network) combined through a learned logistic regression stacking layer — trained on two structurally different real-world pr…
Downloads · 30 days
0
Access
Public
Updated Aug 30, 2026
Repo size
19.4 MB
Likes
1
Public
Click a slice to open those files.
.txt16 MB · 80%
From the Hugging Face model README
Two independent hybrid fraud-detection models — LightGBM (tabular) + GraphSAGE (graph neural network) combined through a learned logistic regression stacking layer — trained on two structurally different real-world proxy datasets, as part of an M.Tech project on cross-channel fraud detection for India's UPI/IMPS digital payment ecosystem.
Framing note: actual large-scale UPI/IMPS transaction data isn't publicly available. These models are trained on real, publicly available proxy datasets (IEEE-CIS, DGraph-Fin) to build and validate the technical architecture for the real target problem — they are not literal UPI/IMPS transaction models, and this is not claimed otherwise.
Full project (web app, data pipeline, source code): https://github.com/KarthickRamAlagar/apexfi-fraud-graph
IEEE-CIS (transaction-level) and DGraph-Fin (account/network-level) have no genuine shared identity or join key between them — combining them into one graph would mean inventing a connection that isn't real. Each dataset is trained and validated independently, using the same hybrid architecture applied to two different real-world structures.
ieee_cis/
├── lightgbm_model.txt # trained LightGBM booster
├── gnn_model.pt # trained GraphSAGE checkpoint
├── stacking_meta.pkl # logistic regression stacker
└── preprocessing_artifacts.pkl # frequency tables, categorical
# encoders, neighbor-lookup index —
# required for scoring genuinely new,
# unseen transactions correctly
dgraph_fin/
├── lightgbm_model.txt
├── gnn_model.pt
└── stacking_meta.pkl
predict_example.py # minimal, runnable loading/inference example
Both models were validated using a fixed, stratified 70/15/15 train/validation/test split, repeated across 3 random seeds, with mean ± standard deviation reported. (This is not k-fold cross- validation — the split itself is fixed; only the model's own training randomness varies across seeds.)
| Model | Precision | Recall | F1 | ROC-AUC |
|---|---|---|---|---|
| LightGBM alone | 0.825 ± 0.005 | 0.768 ± 0.001 | 0.796 ± 0.002 | 0.973 ± 0.000 |
| GraphSAGE alone | 0.466 | 0.522 | 0.492 | 0.892 |
| Stacked (best) | 0.878 ± 0.002 | 0.731 ± 0.001 | 0.798 ± 0.001 | 0.974 ± 0.000 |
Learned stacking weight: ~19:1 favoring LightGBM — tabular features dominate on this dataset; the graph contributes a real but small correction.
| Model | Precision | Recall | F1 | ROC-AUC |
|---|---|---|---|---|
| LightGBM alone | 0.992 ± 0.009 | 0.524 ± 0.000 | 0.686 ± 0.002 | 0.938 ± 0.000 |
| GraphSAGE alone | 1.000 | 0.524 | 0.688 | 0.916 |
| Stacked (best) | 0.995 ± 0.005 | 0.524 ± 0.000 | 0.687 ± 0.001 | 0.938 ± 0.001 |
Learned stacking weight: nearly balanced — unlike IEEE-CIS, the graph and tabular signals are genuinely complementary here.
Research finding worth highlighting: graph information is not universally superior to tabular information — its real, measured value depends on the underlying structure of the specific dataset.
V1-V339 columns cannot be supplied for a
genuinely new transaction (their real-world meaning is undocumented,
even by the dataset's own publishers) — predictions for a new
transaction whose real risk signal lives mostly in those columns will
show an expected, honest gap from what a full-feature retrospective
lookup would show.Research and educational demonstration of hybrid tabular + graph fraud detection architecture, and as a genuine, validated reference point for further work on graph-based fraud detection under full supervision. Not intended for direct production deployment on real financial data without independent validation on that data's actual distribution.
MIT — see the source datasets' own licenses (IEEE-CIS, DGraph-Fin) for any restrictions on the underlying data itself.