Channel randomisation: Self-supervised representation learning for reliable visual anomaly detection in speciality crops

被引:1
|
作者
Choi, Taeyeong [1 ]
Would, Owen [2 ]
Salazar-Gomez, Adrian [2 ]
Liu, Xin [3 ]
Cielniak, Grzegorz [2 ]
机构
[1] Kennesaw State Univ, Dept Informat Technol, 1100 South Marietta Pkwy, Marietta, GA 30060 USA
[2] Univ Lincoln, Lincoln Inst Agrifood Technol, Riseholme Pk LN2 2LG, Lincoln, England
[3] Univ Calif Davis, Dept Comp Sci, 2063 Kemper Hall, Davis, CA 95616 USA
基金
美国国家科学基金会; 美国食品与农业研究所;
关键词
Automated crop monitoring; Non-destructive sensing for quality control; Visual anomaly detection; Data augmentation; Curriculum learning;
D O I
10.1016/j.compag.2024.109416
中图分类号
S [农业科学];
学科分类号
09 ;
摘要
Modern, automated quality control systems for speciality crops utilise computer vision together with a machine learning paradigm exploiting large datasets for learning efficient crop assessment components. To model anomalous visuals, data augmentation methods are often developed as a simple yet powerful tool for manipulating readily available normal samples. State-of-the-art augmentation methods embed arbitrary "structural"peculiarities in normal images to build a classifier of these artefacts (i.e., pretext task), enabling self-supervised representation learning of visual signals for anomaly detection (i.e., downstream task). In this paper, however, we argue that learning such structure-sensitive representations may be suboptimal for agricultural anomalies (e.g., unhealthy crops) that could be better recognised by a different type of visual element like "colour". To be specific, we propose Channel Randomisation (CH-Rand)-a novel data augmentation method that forces deep neural networks to learn effective encoding of "colour irregularities"under self-supervision whilst performing a pretext task to discriminate channel-randomised images. Extensive experiments are performed across various types of speciality crops (apples, strawberries, oranges, and bananas) to validate the informativeness of learnt representations in detecting anomalous instances. Our results demonstrate that CH-Rand's representations are significantly more reliable and robust, outperforming state-of-the-art methods (e.g., CutPaste) that learn structural representations by over 43% in Area Under the Precision-Recall Curve (AUC-PR), particularly for strawberries. Additional experiments suggest that adopting the L*a*b* * a * b * colour space and "curriculum"learning in the pretext task - gradually disregarding channel combinations for unrealistic outcomes - further improves downstream-task performance by 16% in AUC-PR. In particular, our experiments employ Riseholme-2021, , a novel speciality crop dataset consisting of 3.5K real strawberry images gathered in situ from the real farm, along with the Fresh & Stale public dataset. All our code and datasets are made publicly available online to ensure reproducibility and encourage further research in agricultural technologies.
引用
收藏
页数:15
相关论文
共 50 条
  • [21] Self-Supervised Visual Representation Learning via Residual Momentum
    Pham, Trung Xuan
    Niu, Axi
    Zhang, Kang
    Jin, Tee Joshua Tian
    Hong, Ji Woo
    Yoo, Chang D.
    IEEE ACCESS, 2023, 11 : 116706 - 116720
  • [22] Dense Semantic Contrast for Self-Supervised Visual Representation Learning
    Li, Xiaoni
    Zhou, Yu
    Zhang, Yifei
    Zhang, Aoting
    Wang, Wei
    Jiang, Ning
    Wu, Haiying
    Wang, Weiping
    PROCEEDINGS OF THE 29TH ACM INTERNATIONAL CONFERENCE ON MULTIMEDIA, MM 2021, 2021, : 1368 - 1376
  • [23] Self-supervised Representation Learning for Reliable Robotic Monitoring of Fruit Anomalies
    Choi, Taeyeong
    Would, Owen
    Salazar-Gomez, Adrian
    Cielniak, Grzegorz
    2022 IEEE INTERNATIONAL CONFERENCE ON ROBOTICS AND AUTOMATION (ICRA 2022), 2022, : 2266 - 2272
  • [24] Generative and Contrastive Self-Supervised Learning for Graph Anomaly Detection
    Zheng, Yu
    Jin, Ming
    Liu, Yixin
    Chi, Lianhua
    Phan, Khoa T.
    Chen, Yi-Ping Phoebe
    IEEE TRANSACTIONS ON KNOWLEDGE AND DATA ENGINEERING, 2023, 35 (12) : 12220 - 12233
  • [25] Classification-Based Self-Supervised Learning for Anomaly Detection
    Li, Honghu
    Zhu, Yuesheng
    He, Ying
    THIRTEENTH INTERNATIONAL CONFERENCE ON DIGITAL IMAGE PROCESSING (ICDIP 2021), 2021, 11878
  • [26] A NOVEL CONTRASTIVE LEARNING FRAMEWORK FOR SELF-SUPERVISED ANOMALY DETECTION
    Li, Jingze
    Lian, Zhichao
    Li, Min
    2022 IEEE INTERNATIONAL CONFERENCE ON IMAGE PROCESSING, ICIP, 2022, : 3366 - 3370
  • [27] CADet: Fully Self-Supervised Anomaly Detection With Contrastive Learning
    Guille-Escuret, Charles
    Rodriguez, Pau
    Vazquez, David
    Mitliagkas, Ioannis
    Monteiro, Joao
    ADVANCES IN NEURAL INFORMATION PROCESSING SYSTEMS 36 (NEURIPS 2023), 2023,
  • [28] Deep anomaly detection with self-supervised learning and adversarial training
    Zhang, Xianchao
    Mu, Jie
    Zhang, Xiaotong
    Liu, Han
    Zong, Linlin
    Li, Yuangang
    PATTERN RECOGNITION, 2022, 121
  • [29] Pavement anomaly detection based on transformer and self-supervised learning
    Lin, Zijie
    Wang, Hui
    Li, Shenglin
    AUTOMATION IN CONSTRUCTION, 2022, 143
  • [30] Self-Supervised Learning for Anomaly Detection With Dynamic Local Augmentation
    Yoa, Seungdong
    Lee, Seungjun
    Kim, Chiyoon
    Kim, Hyunwoo J.
    IEEE ACCESS, 2021, 9 : 147201 - 147211