Papers
Topics
Authors
Recent
Search
2000 character limit reached

Fraud-Aware Selective CoT Distillation

Updated 6 February 2026
  • The paper introduces a novel fraud-aware selective CoT distillation mechanism that autonomously generates and selects multi-hop reasoning chains to reveal fraud cues in text-attributed graphs.
  • The methodology combines semantic and structural cues through an integrated LLM-GNN pipeline, achieving up to +8.8% improvement in AUPRC over leading baseline models.
  • The approach employs asymmetric co-training and caching strategies to reduce LLM calls and scale batch sizes efficiently, enabling significant computational speedups.

Fraud-aware selective Chain-of-Thought (CoT) distillation is a targeted mechanism developed to enhance graph-based fraud detection in text-attributed graphs (TAGs) by autonomously generating and selecting diverse, semantically-relevant reasoning paths. Integrated within the FraudCoT framework, it addresses the limitations of predefined prompting and decoupled LLM-GNN training pipelines by aligning semantic and structural cues from textual and relational data. This process involves autonomous LLM generation of multi-hop reasoning chains, their selective distillation based on fraud-relevance and diversity, and their integration into node representations for downstream graph neural network (GNN) classification—enabling state-of-the-art detection performance and scalable end-to-end optimization (Li et al., 30 Jan 2026).

1. Problem Formulation and Notation

Fraud-aware selective CoT distillation operates on a text-attributed graph (TAG) G=(V,E,X)G = (V, E, X), where VV is the set of nodes (e.g., reviews, transactions, users), EV×VE \subseteq V \times V represents binary relations, and X={xvvV}X = \{x_v\,|\,v \in V\} comprises raw textual attributes. Each node vv is assigned a binary fraud label yv{0,1}y_v \in \{0, 1\}, and A{0,1}V×VA \in \{0, 1\}^{|V|\times|V|} denotes the adjacency matrix, with N(v)N(v) defining the set of neighbors of vv. The objective is to learn a scoring function fθ(v)f_\theta(v) approximating P(yv=1xv,N(v),)P(y_v=1\,|\,x_v, N(v), \ldots) for accurate node-level fraud detection (Li et al., 30 Jan 2026).

2. Autonomous Chain-of-Thought Generation

The distillation process begins with teacher LLM-based generation of graph-aware reasoning chains for selected nodes. For each distillation node viv_i, a composite prompt pip_i is constructed by concatenating its raw text xix_i and textual descriptions of its 1-hop or multi-hop neighbors {xjjN(vi)}\{x_j \mid j \in N(v_i)\}. The teacher LLM TT receives pip_i and samples SS chain-of-thoughts:

riT,s=T(pi),s=1,,S.r_i^{T,s} = T(p_i), \quad s = 1, \ldots, S.

Each riT,sr_i^{T,s} is a free-form path that references both local semantic features and relational (neighbor) signals, aiming to uncover multi-hop fraud cues embedded in the graph structure (Li et al., 30 Jan 2026).

3. Selective Distillation: Relevance Scoring and Diversity Enforcement

A fraud-aware CoT scoring function s(r;G)s(r;G) is proposed to select high-value chains:

s(r;G)=αsim(Enctext(r),Enctext(xi))+(1α)jN(vi)sim(Enctext(r),Enctext(xj)),s(r;G) = \alpha\,\text{sim}(\text{Enc}_\text{text}(r), \text{Enc}_\text{text}(x_i)) + (1-\alpha)\, \sum_{j\in N(v_i)} \text{sim}(\text{Enc}_\text{text}(r), \text{Enc}_\text{text}(x_j)),

where sim(,)\text{sim}(\cdot,\cdot) is cosine similarity and Enctext\text{Enc}_\text{text} is a lightweight encoder (e.g., BERT-tiny), balancing semantic relevance and structural alignment. Scores are normalized to [0,1][0,1] across samples.

To ensure that distilled CoTs are both fraud-relevant and diverse, the procedure selects:

  • SposS_{\text{pos}}: the top-kk CoTs with scores τpos\ge \tau_{\text{pos}},
  • SnegS_{\text{neg}}: the bottom-kk CoTs with scores τneg\le \tau_{\text{neg}}, retaining only those with pairwise Levenshtein edit distance D\ge D to enforce diversity.

The student model SS is distilled by minimizing:

Ldistill=ErP[logPS(rpi)]+λErN[log(1PS(rpi))],\mathcal{L}_\text{distill} = \mathbb{E}_{r\in P}[-\log P_S(r|p_i)] + \lambda\,\mathbb{E}_{r\in N}[-\log(1-P_S(r|p_i))],

where PP and NN are sets of positive and negative chains, and λ\lambda is an unlikelihood loss coefficient. LoRA (Low-Rank Adaptation) adapters with rank r=8r=8–$16$ are employed for efficient parameterization during student finetuning. Empirically, U300U \approx 300 nodes, S=5S=5 samples, k=2k=2–3, and λ100\lambda \approx 100 are optimal (Li et al., 30 Jan 2026).

4. Integration of Distilled CoTs into GNN Representation

After student finetuning, a distilled CoT is generated for each node vv as rv=S(xv,xN(v))r_v = S(x_v, x_{N(v)}) and concatenated with the raw text to yield xv=[xvrv]x'_v = [x_v \oplus r_v]. This extended textual attribute is processed by an LLM-style encoder Encθ\text{Enc}_\theta to obtain initial node embeddings:

hv(0)=Encθ(xv)h_v^{(0)} = \text{Enc}_\theta(x'_v)

These enriched embeddings encode both direct and multi-hop semantic-structural fraud cues, providing a more informative input for the downstream GNN (Li et al., 30 Jan 2026).

5. Asymmetric LLM–GNN Co-Training and Efficiency

A key component is the asymmetric co-training pipeline. In each batch BB, only target nodes viv_i are freshly encoded:

hi(0)=Encθ(xi)h_i^{(0)} = \text{Enc}_\theta(x'_i)

while neighbor nodes jN(vi)j \in N(v_i) utilize cached embeddings hˉj(0)\bar h_j^{(0)} computed once at initialization. Message passing employs LL heterogeneous GraphSAGE layers:

hi(l+1)=σ(Wr(l)mean({hi(l)}{hj(l):jNr(vi)}))h_i^{(l+1)} = \sigma\left( W_r^{(l)}\cdot \text{mean}\left( \{ h_i^{(l)} \} \cup \{ h_j^{(l)} : j \in N_r(v_i) \} \right) \right)

The total loss is a weighted sum of binary cross-entropy node classification and distillation loss:

L=αLGNN+(1α)Ldistill\mathcal{L} = \alpha\,\mathcal{L}_\text{GNN} + (1-\alpha)\,\mathcal{L}_\text{distill}

Typically, α=0.8\alpha=0.8. Only target nodes propagate gradients through Encθ\text{Enc}_\theta, greatly reducing backpropagation cost.

In terms of efficiency, the asymmetric strategy decreases per-epoch LLM calls from O(NKHL(TD2+T2D))O(NK^HL(TD^2+T^2D)) (naive) to O(N)O(N), with a one-time O(N)O(N) cache build. Using the DigitalMusic dataset, this yields epoch time reduction from 58,63958{,}639 s (naive) to $55$ s (asymmetric), corresponding to a 1,066×1{,}066 \times speedup, and permits batch size scaling from $2$ to $256$ (Li et al., 30 Jan 2026).

6. Empirical Evaluation and Baselines

FraudCoT, incorporating fraud-aware selective CoT distillation, is experimentally validated on InstantVideo (37,126 nodes, 9.9M edges), DigitalMusic (64,706 nodes, 7.7M edges), and PromotionAbuse (371K nodes, 1.3M edges). Comparisons are drawn against:

  • GNNs: GraphSAGE, HGT, ConsisGAD, PMP, GAAP
  • Pure LLMs: Qwen-8B, InstructGLM
  • Graph-enhanced LLMs: LLaGA, GraphGPT, HiGPT
  • LLM-enhanced GNNs: TAPE, FLAG

Evaluation metrics include Macro-F1, AUROC, and AUPRC (using scikit-learn implementations). FraudCoT achieves up to +8.8%+8.8\% improvement in AUPRC over the best baseline (PromotionAbuse), with consistent performance gains across all public and industrial dataset splits. Ablation studies confirm that both the selective distillation and asymmetric co-training components are essential for optimal results (Li et al., 30 Jan 2026).

Dataset Nodes Edges Best AUPRC Gain
InstantVideo 37,126 9.9M up to +8.8%
DigitalMusic 64,706 7.7M up to +8.8%
PromotionAbuse 371,000 1.3M up to +8.8%

7. Implementation Strategies and Practical Considerations

Implementation guidelines include:

  • Use LoRA adapters with rank r=8r=8–$16$.
  • Set unlikelihood loss λ100\lambda \approx 100.
  • Sample K=10K=10 neighbors, with hop number H=2H=2.
  • Cache node embeddings on CPU or host memory to minimize GPU LLM calls.
  • Apply early stopping based on validation AUROC; batch sizes $128$–$256$ provide optimal throughput.
  • Precompute all text encodings for CoT scoring to accelerate distillation selection.

A plausible implication is that this design allows practitioners to scale LLM-enhanced GNNs to industrial TAGs with both superior predictive performance and orders-of-magnitude throughput improvements, without requiring expensive end-to-end LLM inference on all neighbors during training (Li et al., 30 Jan 2026).

Definition Search Book Streamline Icon: https://streamlinehq.com
References (1)

Topic to Video (Beta)

No one has generated a video about this topic yet.

Whiteboard

No one has generated a whiteboard explanation for this topic yet.

Follow Topic

Get notified by email when new papers are published related to Fraud-Aware Selective CoT Distillation.