arXiv paper· database systems (learned indexes, adversarial machine learning)· method paper

Poisoning Attacks on the PGM-index

Atsuki Sato, Martin Aumüller, Yusuke Matsui

cs.DB · cs.CR · cs.LG

Quick verdict

Theory-backed PGM-attack inflates PGM-index segments up to 120x with a 52%-of-optimum certificate; rigorous, but query-time impact stays modest (max 1.22x).

Summary

The paper proposes PGM-attack, a white-box poisoning attack against the optimal piecewise linear approximation (PLA) underlying the PGM-index and other PLA-based learned indexes. Before index construction, the attacker inserts up to λ keys inside the legitimate key range to inflate m_opt, the minimum number of PLA segments, which dominates PGM-index size. The authors build bottom-up from three problems: maximizing the maximum error of a linear segment (P1), minimizing covered legitimate keys (P2, reduced to P1 via Theorem 4), and maximizing segment count (P3). Structural theorems show optimal poisons form blocks adjacent to legitimate keys; DI-Consecutive evaluates candidates in O(|I| log c) via discrete intercepts and sparse tables. An instance-dependent upper bound (block relaxation, Monge DAG shortest paths, Lagrangian relaxation, concave knapsack allocation) certifies the attack reaches at least 52% of the optimum on every instance (bound ≤1.92× the attack result). With λ=0.1n and ε=128, segment count grows up to 120× (YCSB: 663→79.8K), PGM index size up to 120× and 1.46–6.73× on other real datasets; query time up to 1.22×. The attack transfers to FITing-Tree (up to 18.5×) and PGM++ (up to 35,770× on YCSB), while B+-trees are essentially unaffected. Limitations: impact is concentrated in memory rather than latency, the greedy heuristic's optimality gap is undecomposed, generation takes 1.0–14.7 hours, and no defense is evaluated.

Review complete

Assessment at a glance

database systems (learned indexes, adversarial machine learning)

4

risk flags identified

● high 0

● medium 2

● low 2

Reproducibility readinesshigh
Code & data availableyes
Review confidencehigh
AI text signal1/20 passages flagged
Deep research viableCode-reproducible

5 strengths · 5 weaknesses · 5 author questions

AI text signal = Binoculars zero-shot detector score over 20 sampled passages (Fast-DetectGPT co-reported) — absence does not prove human authorship, and a flag does not imply misconduct.

Levels reflect verification complexity — not a quality score. Evolve never rates your paper.

Key claims under review

  • Inserting poison keys equal to 10% of the dataset before construction inflates the optimal PLA segment count by up to 120× (YCSB, ε=128) and the PGM-index size accordingly.
  • An instance-dependent upper bound is within 1.92× of PGM-attack's segment count on all evaluated instances, certifying the attack reaches at least 52% of the optimal amplification.
  • The attack transfers to other PLA-based learned indexes (FITing-Tree up to 18.5×, PGM++ up to 35,770× on YCSB) while B+-trees remain essentially unaffected.
  • The vulnerability is intrinsic to the minimum-segment PLA optimization objective itself, not to PGM-specific implementation, motivating robustness-aware objectives for future learned indexes.

Strengths

  • Proven optimality certificate

    The instance-dependent upper bound is at most 1.92× PGM-attack's segment count on every instance in Table 5, mathematically certifying ≥52% of the optimum; Theorems 5–6 additionally prove the instance-agnostic bound (m_opt(K)+|P|) tight, which justifies the heavier instance-specific machinery.

  • Layered theory with real leverage

    Theorem 1 (optimal poisons form blocks adjacent to legitimate keys) cuts the search space from universe-size to O(C(2n−3+λ, λ)); Theorem 2 proves duplicate-allowed optima concentrate on a single legitimate key; Theorem 4 reduces P2 to P1 over key prefixes — each carries a full proof or proof sketch (Appendices A.1–A.4).

  • Scalable algorithms evaluated at genuine scale

  • Breadth and fairness of evaluation

  • Honest threat-model scoping

Weaknesses

  • Memory-centric headline; query impact small

  • Undecomposed optimality gap

  • No defense implemented or evaluated

  • Thin dispersion reporting

  • Generation cost and static setting

Full Review Report

$1.99 · one-time

Unlock the complete preliminary analysis Evolve produced for this paper.

Report preview

Scope & Claims

This is a method paper at the intersection of learned indexes and adversarial machine learning: it asks whether the *optimal* PLA at the heart of the PGM-index (Ferragina & Vinciguerra, VLDB'20) is itself poisonable. The authors claim three contributions: (1) a layered problem formulation — P1 (maximize the maximum error of one linear segment), P2 (minimize the legitimate keys a segment covers), P3 (maximize total PLA segments) — with structural theorems for each; (2) PGM-attack, a fast sequential heuristic for P3; (3) efficiently computable upper bounds on the post-attack m_opt, the minimum segment count that dominates PGM-index size and query structure (§2.2). Headline numbers: λ=0.1n poison keys inflate m_opt by up to 120× (YCSB, ε=128: 663→79.8K, Table 5) and PGM index size by up to 120× (Table 4).

The remaining sections cover Methodology Assessment, Statistical Evidence, Results & Reproducibility, Limitations & Risks, and Verdict & Recommendations.

  • All 5 strengths & 5 weaknesses, analyzed in detail
  • 4 risk flags with severity, section & reviewer questions
  • Methods & statistical evidence assessment
  • 5 questions a peer reviewer would ask
  • Full six-dimension evaluation report & recommendations

One-time purchase · this paper only

One-time purchase, this paper only. AI-generated preliminary analysis — not peer review.

Frequently asked questions

What is a poisoning attack on the PGM-index?

An adversary inserts specially chosen keys into the dataset before the index is built. The PGM-index approximates the key-to-rank (CDF) mapping with the minimum number of linear segments (an optimal PLA); PGM-attack places poison keys where they create many hard-to-approximate regions, multiplying the required segment count and thus the index's memory footprint.

How much damage can PGM-attack do?

With poison keys equal to 10% of the dataset (ε=128), segment count and index size grow up to 120× on YCSB and 1.46–6.73× on other real datasets, with query time up to 1.22× slower. Even 1% poisoning yields up to 15.5× on YCSB. A B+-tree, by contrast, grows only ~1.1× and is essentially unaffected.

Does the attacker need to know the index configuration?

The base attack is white-box (knows all legitimate keys and the error parameter ε), but Table 6 shows poison keys generated with a wrong ε guess still work: on YCSB with ε_true=128, even 8×-mismatched attacks amplify segment count by at least 22.7×, so the attack also functions as a gray-box attack.

Does the attack work on other learned indexes besides PGM?

Yes, especially PLA-based ones: under 10% poisoning at ε=128, index size grows up to 18.5× for FITing-Tree, up to 35,770× for PGM++ (whose legitimate YCSB index is exceptionally small), and around 98.6× for ALEX on YCSB. RMI keeps a fixed model size but its query time rises up to 1.33×; B+-trees and ART are largely unaffected.

About this review

This is an AI-generated preliminary review by Evolve.Pub, produced from the openly available text of the paper. It is not peer review and does not represent the authors' views. The canonical record of the paper is at the publisher link above.

  • The provided text is a sampled excerpt with [...sampling gap...] regions (e.g., parts of Sections 3.2–6.4 and Appendices C–D were skipped); all figures, tables and numbers cited in this review are taken from the visible portions only.
  • The complete DAG-construction portion of the instance-dependent bound proof (Appendix C.2, immediately after Lemma 2) fell inside a sampling gap, so that argument is assessed from the visible description and proof sketches rather than in full.

Go deeper on this paper

Wrote a paper? Stress-test it before reviewers do.

Upload your manuscript for the same structured scrutiny — reasoning, statistics, evidence, figures, and reproducibility.

Scan your paper
Poisoning Attacks on the PGM-index | Evolve.Pub