Skip to content
AI News HubLIVE
Original source2 min read

Don't Just Impute, Detect: Task-Aware Diffusion Pipelines for Edge Anomaly Detection

Summary

The paper identifies an "imputation bias" in the standard impute-then-detect pipeline for IIoT edge data loss and proposes the Task-Aware Diffusion Refiner (TADR), which refines imputations against a frozen downstream detector's loss, raising F1 by 0.338 on SMD and characterizing when the method helps and how fast it runs on edge hardware.

SourcearXiv Machine LearningAuthor: Jeongsu Park, Taehyeon Yun, Geonwoo Kim, Yuji Lim, Moohong Min
Don't Just Impute, Detect: Task-Aware Diffusion Pipelines for Edge Anomaly Detection
Report an error

The correction channel is not available yet. You can copy the article reference below for later.

Correction instructions
Read article

[Submitted on 29 Oct 2026]

Title:Don't Just Impute, Detect: Task-Aware Diffusion Pipelines for Edge Anomaly Detection

View a PDF of the paper titled Don't Just Impute, Detect: Task-Aware Diffusion Pipelines for Edge Anomaly Detection, by Jeongsu Park and 4 other authors

View PDF HTML (experimental)

Abstract:Industrial Internet of Things (IIoT) edge pipelines routinely lose contiguous blocks of multi-sensor data to network dropouts and hardware faults. The standard remedy is a disjointed "impute-then-detect" pipeline whose imputer is trained only for Mean Squared Error (MSE). Such an imputer fills a missing block with the most probable normal pattern: a partially observed anomaly is smoothed back toward normality, and boundary artifacts on normal data raise false alarms. We call this imputation bias and propose the Task-Aware Diffusion Refiner (TADR), a conditional diffusion refiner trained against the loss landscape of a frozen downstream detector. Because back-propagating through the sampling trajectory is intractable, the task-aware terms act on the one-step clean-window estimate that the noise prediction already yields, so training costs no more than plain diffusion training. Under strict Point-to-Point evaluation over five seeds with paired significance tests, TADR raises F1 by $0.338$ over the unrefined detector on SMD ($p = 0.0015$), and the anomaly enhancement objective alone contributes $0.463$ in ablation. We also characterise when the method helps: the gain tracks the continuous, mutually redundant channels available and the anomaly evidence the outage removes, and it is absent on telemetry datasets dominated by one-hot command channels and on PSM. At ten sampling steps TADR runs in $41.4$ ms per window on a single CPU core, sustaining about $24$ concurrent 1 Hz streams from a $4.03$ MB model. Code and per-seed results are available at this https URL.

Comments: 11 pages, 6 figures. Accepted at the 33rd IEEE International Conference on High Performance Computing, Data, and Analytics (HiPC 2026)

Subjects:

Machine Learning (cs.LG)

Cite as: arXiv:2610.40732 [cs.LG]

(or arXiv:2610.40732v1 [cs.LG] for this version)

https://doi.org/10.48550/arXiv.2610.40732

arXiv-issued DOI via DataCite (pending registration)

Submission history

From: Geonwoo Kim [view email] [v1] Thu, 29 Oct 2026 12:23:18 UTC (1,762 KB)

Full-text links:

Access Paper:

View a PDF of the paper titled Don't Just Impute, Detect: Task-Aware Diffusion Pipelines for Edge Anomaly Detection, by Jeongsu Park and 4 other authors

View PDF

HTML (experimental)

TeX Source

view license

Additional Features

Audio Summary

Current browse context:

cs.LG

new | recent | 2026-10

Change to browse by:

cs

References & Citations

SciX

Google Scholar

Semantic Scholar

Loading...

Data provided by:

Bibliographic Tools

Bibliographic and Citation Tools

Bibliographic Explorer Toggle

Bibliographic Explorer (What is the Explorer?)

Connected Papers Toggle

Connected Papers (What is Connected Papers?)

Litmaps Toggle

Litmaps (What is Litmaps?)

scite.ai Toggle

scite Smart Citations (What are Smart Citations?)

Code, Data, Media

Code, Data and Media Associated with this Article

alphaXiv Toggle

alphaXiv (What is alphaXiv?)

Links to Code Toggle

CatalyzeX Code Finder for Papers (What is CatalyzeX?)

DagsHub Toggle

DagsHub (What is DagsHub?)

GotitPub Toggle

Gotit.pub (What is GotitPub?)

Huggingface Toggle

Hugging Face (What is Huggingface?)

ScienceCast Toggle

ScienceCast (What is ScienceCast?)

Demos

Demos

Replicate Toggle

Replicate (What is Replicate?)

Spaces Toggle

Hugging Face Spaces (What is Spaces?)

Spaces Toggle

TXYZ.AI (What is TXYZ.AI?)

Related Papers

Recommenders and Search Tools

Link to Influence Flower

Influence Flower (What are Influence Flowers?)

Core recommender toggle

CORE Recommender (What is CORE?)

IArxiv recommender toggle

IArxiv Recommender (What is IArxiv?)

Author

Venue

Institution

Topic

About arXivLabs

arXivLabs: experimental projects with community collaborators

arXivLabs is a framework that allows collaborators to develop and share new arXiv features directly on our website.

Both individuals and organizations that work with arXivLabs have embraced and accepted our values of openness, community, excellence, and user data privacy. arXiv is committed to these values and only works with partners that adhere to them.

Have an idea for a project that will add value for arXiv's community? Learn more about arXivLabs.

Which authors of this paper are endorsers? | Disable MathJax (What is MathJax?)

Key points and analysis

Article intelligence

InvestorsAdvanced

Key points

  • MSE-trained imputers smooth partially observed anomalies back toward normality and create boundary artifacts that trigger false alarms.
  • TADR trains a conditional diffusion refiner against a frozen detector's loss, acting on the one-step clean-window estimate so training cost matches plain diffusion.
  • On SMD, TADR raises F1 by 0.338 over five seeds (p = 0.0015), and the anomaly enhancement objective alone contributes 0.463 in ablation.
  • At ten sampling steps it runs in 41.4 ms per window on one CPU core, sustaining about 24 concurrent 1 Hz streams from a 4.03 MB model.

Highlights and analysis are generated automatically and may contain errors. Check the original source.