Link prediction is a core graph machine learning task that computes the likelihood of an edge existing between two nodes, either because the edge is unobserved in incomplete data or because it will form in the future. In a citation graph, nodes represent legal cases and edges represent citations; link prediction algorithms analyze the network's structural patterns—such as common neighbors, preferential attachment, and node embeddings—to forecast which authorities a judge or litigant will reference in a forthcoming opinion.
Glossary
Link Prediction

What is Link Prediction?
Link prediction is a machine learning task that estimates the probability of a missing or future connection forming between two nodes in a network, applied to legal citation graphs to forecast which precedents a court is likely to cite.
Modern legal AI systems implement link prediction using Graph Neural Networks (GNNs) that learn latent representations of cases by aggregating features from their citation neighborhoods. These models outperform heuristic methods by capturing non-linear dependencies, such as the tendency of courts to cite cases that share semantic similarity and occupy structurally proximate positions in the authority graph. The output is a ranked list of candidate precedents, enabling proactive citation recommendation and the early detection of shifts in precedential weight before they become explicit in published decisions.
Core Characteristics of Link Prediction
The fundamental computational properties and algorithmic approaches that define how machine learning models infer missing or future connections within legal citation networks.
Heuristic Scoring Methods
Classical approaches that compute connection likelihood based on topological proximity without requiring model training.
- Common Neighbors: Scores node pairs by the number of shared adjacent nodes, assuming cases cited by the same authorities are related.
- Jaccard Coefficient: Normalizes common neighbors by total neighborhood size to avoid bias toward high-degree nodes.
- Adamic-Adar: Weights shared neighbors inversely by their degree, prioritizing rare connections over ubiquitous citations.
- Preferential Attachment: Predicts links based on the product of node degrees, modeling the tendency of highly-cited cases to attract future citations.
Node Embedding Techniques
Methods that learn low-dimensional vector representations of legal nodes by preserving graph structure, enabling similarity-based link prediction.
- Node2Vec: Uses biased random walks to capture both homophily and structural equivalence in citation patterns.
- DeepWalk: Treats truncated random walks as sentences, applying skip-gram models to learn latent representations of cases.
- Spectral Embedding: Decomposes the graph Laplacian matrix to position nodes in a space where connected cases are proximate.
- Embedding distance or dot product between two case vectors serves as the link probability score.
Graph Neural Network Approaches
Deep learning architectures that generate node representations by recursively aggregating features from local citation neighborhoods.
- Graph Convolutional Networks (GCNs) apply spectral convolutions to learn node embeddings that capture multi-hop citation dependencies.
- GraphSAGE samples and aggregates features from a node's local neighborhood, enabling inductive prediction for newly published cases.
- Graph Attention Networks (GATs) assign learnable importance weights to different citing authorities during neighborhood aggregation.
- An edge decoder—typically a multi-layer perceptron or dot product—computes the final link probability from the learned node pair embeddings.
Temporal Dynamics Modeling
Techniques that incorporate the time dimension to predict future citations rather than merely recovering missing historical links.
- Temporal Edge Features: Encodes the age difference between cases and the recency of prior citations as input signals.
- Time-Aware Negative Sampling: Selects negative edges from the same time window as positive edges to prevent temporal leakage during training.
- Recurrent Graph Architectures: Combines GNNs with RNNs or temporal attention to model the evolution of citation networks over sequential time snapshots.
- Hawkes Process Integration: Models citation events as a self-exciting point process where each citation temporarily increases the probability of subsequent citations.
Evaluation Metrics
Quantitative measures used to assess link prediction performance, critical given the extreme class imbalance in sparse citation graphs.
- Area Under ROC Curve (AUC-ROC) measures the trade-off between true positive and false positive rates across thresholds.
- Mean Reciprocal Rank (MRR) evaluates how highly the correct missing link is ranked among all candidate edges.
- Hits@K measures the fraction of true links appearing in the top-K predictions, with K values of 10, 50, or 100 being standard.
- Evaluation requires temporal splitting: training on citations before a cutoff date and testing on citations formed after that date to simulate real-world forecasting.
Feature Engineering for Legal Graphs
Domain-specific input signals that enhance link prediction beyond pure topology by encoding jurisprudential knowledge.
- Court Hierarchy Distance: Encodes the vertical separation between the issuing courts of two cases within the judicial system.
- Subject Matter Overlap: Measures the semantic similarity of legal topics or West Key Number classifications between cases.
- Treatment Sentiment Features: Incorporates the polarity of existing citations as edge weights, distinguishing supportive from critical authority chains.
- Jurisdictional Concordance: A binary or scalar feature indicating whether two cases originate from the same sovereign jurisdiction, a strong predictor of citation likelihood.
Enabling Efficiency, Speed & Accuracy
Intelligent Analysis, Decision & Execution
We build AI systems for teams that need search across company data, workflow automation across tools, or AI features inside products and internal software.
Talk to Us
Search across company data
Give teams answers from docs, tickets, runbooks, and product data with sources and permissions.
Useful when people spend too long searching or get different answers from different systems.

Automate internal workflows
Use AI to route work, draft outputs, trigger actions, and keep approvals and logs in place.
Useful when repetitive work moves across multiple tools and teams.

Add AI to products and internal tools
Build assistants, guided actions, or decision support into the software your team or customers already use.
Useful when AI needs to be part of the product, not a separate tool.
Frequently Asked Questions
Explore the core concepts behind link prediction in legal citation networks, a critical machine learning task for forecasting precedent relationships and building intelligent authority graphs.
Link prediction is a machine learning task that estimates the probability of a missing or future edge between two nodes in a graph. In the context of a citation network, it forecasts which precedents a court is likely to cite in a new opinion or identifies missing citations that should exist between semantically related cases. The model analyzes the existing graph topology—including co-citation patterns, bibliographic coupling, and node attribute similarity—to score candidate links. This capability powers citation recommendation systems for legal drafters and helps maintain the structural integrity of authority graphs by detecting gaps in the network.
Related Terms
Master the computational mapping and traversal of legal authority graphs with these foundational concepts essential for building precedent intelligence systems.
Citation Graph
A directed network structure where nodes represent legal cases or statutes and edges represent citation relationships. This forms the foundational data structure for computational precedent analysis.
- Nodes: Individual legal authorities (cases, statutes, regulations)
- Edges: Directed citations from a citing case to a cited authority
- Edge attributes: May encode treatment type, citation intent, or temporal metadata
- Scale: Modern legal graphs contain millions of nodes and tens of millions of edges
Citation graphs enable graph traversal algorithms to map the flow of legal reasoning across jurisdictions and time periods.
Authority Propagation
A graph algorithm that iteratively distributes precedential influence scores across a citation network, often using PageRank variants, to identify the most legally significant nodes.
- Mechanism: Nodes pass authority weight to their out-neighbors proportionally
- Convergence: Algorithm iterates until score distribution stabilizes
- Damping factor: Prevents rank sinks and models random judicial discovery
- Weighted variants: Incorporate treatment sentiment and court hierarchy into edge weights
Authority propagation surfaces landmark cases that may not have the highest raw citation count but sit at critical junctures in the legal reasoning network.
Treatment Type Classification
An NLP task that automatically categorizes how a citing case legally treats a cited authority, assigning labels such as:
- Overruled: Explicit invalidation of the prior holding
- Distinguished: Material factual or legal differences found
- Followed: Precedent applied as controlling authority
- Criticized: Negative commentary without explicit overruling
- Harmonized: Reconciled with apparently conflicting authority
Accurate treatment classification is essential for maintaining citation graph integrity and computing reliable authority scores that reflect actual judicial reception rather than mere citation frequency.
Graph Neural Network (GNN)
A deep learning architecture designed to operate directly on graph-structured data, used in legal AI to learn node embeddings that capture both a case's intrinsic features and its citation neighborhood structure.
- Message passing: Nodes aggregate feature information from their neighbors iteratively
- Node embeddings: Dense vector representations encoding both content and graph position
- Applications: Link prediction, node classification, and authority score refinement
- Architectures: Graph Convolutional Networks (GCNs), Graph Attention Networks (GATs), and GraphSAGE
GNNs enable link prediction models to learn complex, non-linear patterns in citation behavior that simple heuristic metrics cannot capture.
Temporal Citation Analysis
The study of citation patterns over time to model how legal authority evolves, ages, or gains influence, incorporating timestamps into graph models to detect trends like precedent aging.
- Citation velocity: Rate at which a decision accumulates citations post-publication
- Precedent aging: Decline in citation frequency as doctrines mature or are superseded
- Temporal link prediction: Forecasting future citations based on historical patterns
- Epoch encoding: Time-aware node embeddings that capture a case's historical context
Temporal analysis distinguishes between a case that is historically significant but no longer actively cited and one experiencing a resurgence in relevance due to new doctrinal developments.
Betweenness Centrality
A graph metric measuring how often a node lies on the shortest path between other nodes, used in citation networks to identify cases that serve as critical bridges connecting distinct doctrinal clusters.
- Computation: Fraction of all shortest paths in the graph that pass through the node
- Interpretation: High betweenness cases connect otherwise disconnected areas of law
- Bridge detection: Identifies cases that synthesize multiple doctrinal threads
- Vulnerability analysis: Removal of high-betweenness nodes fragments the citation graph
Cases with high betweenness centrality often represent seminal decisions that unified previously fragmented areas of jurisprudence, making them critical targets for link prediction models.

About the author
Prasad Kumkar
CEO & MD, Inference Systems
Prasad Kumkar is the CEO & MD of Inference Systems and writes about AI systems architecture, LLM infrastructure, model serving, evaluation, and production deployment. Over 5+ years, he has worked across computer vision models, L5 autonomous vehicle systems, and LLM research, with a focus on taking complex AI ideas into real-world engineering systems.
His work and writing cover AI systems, large language models, AI agents, multimodal systems, autonomous systems, inference optimization, RAG, evaluation, and production AI engineering.
Partnered with leading AI, data, and software stack.
How We Work
Custom AI workflows for your Business
One-fit-all AI don't work for modern businesses. At Inferensys, we aim to understand your business & custom requirements; which we use to define most efficient agentic workflows, the data, and the tools for your business.
01
Review the use case
We understand the task, the users, and where AI can actually help.
Read more02
Pick the right approach
We define what needs search, automation, or product integration.
Read more03
Build the first useful version
We implement the part that proves the value first.
Read more04
Improve from there
We add the checks and visibility needed to keep it useful.
Read moreThe first call is a practical review of your use case and the right next step.
Talk to Us