LeonX995 / MTAGNView external linksLinks
The PyTorch implementation for multi-task attention guided network (MTAGN) in End to End Multi-task learning with Attention for Multi-objective Fault Diagnosis under Small Sample
☆25Dec 6, 2023Updated 2 years ago
Alternatives and similar repositories for MTAGN
Users that are interested in MTAGN are comparing it to the libraries listed below
Sorting:
- The source code of the model: Multi-Scale Graph Convolutional Network with Contrastive-Learning enhanced Self-attention Pooling (MSGCN-CS…☆18Apr 10, 2024Updated last year
- The source code (Pytorch version) of paper "Multi-modality augmented Prototypical Network for Fault Diagnosis"☆11Aug 26, 2024Updated last year
- The PyTorch version for Semi-supervised meta-learning networks with squeeze-and-excitation attention for few-shot fault diagnosis.☆61Dec 1, 2021Updated 4 years ago
- A Dual-View Style Mixing Network for Unsupervised Cross-Domain Fault Diagnosis With Imbalanced Data☆23Mar 18, 2025Updated 10 months ago
- A fault diagnosis method for rotating machinery based on CNN with mixed information☆34Oct 19, 2023Updated 2 years ago
- ACAN: A Plug-and-Play Adaptive Center-Aligned Network for Unsupervised Domain Adaptation☆21Sep 12, 2024Updated last year
- The source codes of Meta-learning for few-shot cross-domain fault diagnosis.☆178Feb 16, 2025Updated 11 months ago
- GTFE-Net: A Gramian Time Frequency Enhancement CNN for bearing fault diagnosis☆40Mar 27, 2023Updated 2 years ago
- MSIFT: A Novel End-to-End Mechanical Fault Diagnosis Framework under Limited & Imbalanced Data Using Multi-Source Information Fusion☆66Feb 26, 2025Updated 11 months ago
- Capsule network for fault diagnosis （胶囊网络用于故障诊断）☆34May 16, 2023Updated 2 years ago
- Codes of paper "A Novel Framework Based on Adaptive Multi-Task Learning for Bearing Fault Diagnosis".☆23Apr 28, 2023Updated 2 years ago
- Use Paderborn data and HIilbert theory to diagnosis the fault of rolling bear with MATLAB.☆23Jun 10, 2025Updated 8 months ago
- This repository is for the Few-shot Learning for the fault diagnosis of large industrial equipment.☆98Jun 9, 2022Updated 3 years ago
- The repository of the paper 'AWD-Net: Adaptive Wavelet Decomposition Network for Rotating Machinery Fault Diagnosis Under Noisy Environme…☆20Nov 24, 2023Updated 2 years ago
- 实现的是WDCNN的pytorch版本代码，对应论文的第三章 data包含了四个数据文件夹，这里只使用了0HP文件夹中的数据，里面包含了正常、内圈、外圈、滚动体共10种状态 preprocess.py的功能是对数据进行采样、编码，虽然划分出来了验证集但是并没有使用 trai…☆27Jan 11, 2023Updated 3 years ago
- [MSSP 2022] A domain generalization network combing invariance and specificity towards real-time intelligent fault diagnosis☆15Apr 1, 2024Updated last year
- The code of Interpretable Convolutional Neural Network with Multilayer Wavelet for Noise-Robust Machinery Fault Diagnosis☆51Jul 6, 2022Updated 3 years ago
- Source code for the paper "Evaluating calibration of deep fault diagnostic models under distribution shift" published in Journal Computer…☆20Jul 3, 2025Updated 7 months ago
- 基于图神经网络的机械故障诊断☆107Feb 27, 2024Updated last year
- ☆13Sep 11, 2022Updated 3 years ago
- Open dataset in the field of mechanical fault diagnosis under variable speed conditions, providing benchmark for algorithm performance ev…☆33Nov 1, 2025Updated 3 months ago
- Code tu use learnable wavelet transforms like L-WPT and DeSPaWN methods in pytorch☆33Feb 2, 2023Updated 3 years ago
- TFSCL☆19Jul 25, 2024Updated last year
- Dual Adversarial and Contrastive Network for Single-Source Domain Generalization in Fault Diagnosis☆17Mar 7, 2025Updated 11 months ago
- A Dual-View Network For Imbalanced Fault Diagnosis of Rotating Machinery☆11Aug 29, 2023Updated 2 years ago
- A Multi-source Cross-speed Bearing Fault Diagnosis Method☆12Jan 7, 2025Updated last year
- Importance-aware Subgraph Convolutional Networks Based on Multi-source Information Fusion for Cross-domain Mechanical Fault Diagnosis☆50Feb 26, 2025Updated 11 months ago
- The intelligent fault diagnosis of HNU IDG☆121Nov 9, 2022Updated 3 years ago
- 智能故障诊断中一维类梯度激活映射可视化展示 1D-Grad-CAM for interpretable intelligent fault diagnosis☆115May 16, 2023Updated 2 years ago
- ☆36Nov 29, 2022Updated 3 years ago
- [EAAI 2024] A federated distillation domain generalization framework for machinery fault diagnosis with data privacy☆13Apr 3, 2024Updated last year
- Live Editor tutorials for MATLAB deep learning toolkit CAM explainability analysis☆12Nov 16, 2022Updated 3 years ago
- ☆14Jan 25, 2025Updated last year
- Code for IJCAI'24 paper: Where to Mask: Structure-Guided Masking for Graph Masked Autoencoders☆14Apr 30, 2024Updated last year
- ☆16Jul 21, 2022Updated 3 years ago
- Code repository for Fault diagnosis of rotating machinery based on recurrent neural networks☆16Mar 15, 2022Updated 3 years ago
- 基于Laplace小波卷积和BiGRU的少量样本故障诊断方法 (Small sample fault diagnosis based on Laplace wavelet convolution and BiGRU)☆68Jun 10, 2025Updated 8 months ago
- Produce an example using LSTM to predict remaining useful life of machinery☆16Dec 28, 2023Updated 2 years ago
- ☆15Nov 29, 2024Updated last year