Scientific Machine LearningGeometric Deep LearningBayesian Deep LearningMiscellaneous

Bayesian Deep Learning

Bayesian deep learning is an approach that designs deep learning based on Bayesian modeling. In particular, deep generative models represent causality and dependency among observations using a graph structure called a Bayesian network, and then implement those relationships with deep learning. This allows us to visualize the interpretable rationale behind outputs, quantify the reliability (or uncertainty) of decision-making, and efficiently analyze small datasets. It is also part of the foundational technology behind generative AI.

Bayesian Deep Learning

Deep Generative Models

Deep Generative Models for Medical Diagnosis Using Brain Functional Imaging

Brain functional imaging is expected to supplement the diagnosis of mental disorders based on clinical interviews by providing objective criteria. However, data collection is extremely costly, and existing datasets are very small at the scale required for standard deep learning. Furthermore, individual variability such as age and gender often hinders detecting disease-related activity patterns. Additionally, discrepancies in imaging equipment across hospitals can cause further complications. To address these issues, we construct a Bayesian network using a deep generative model that simultaneously captures both the disease-related activity patterns and individual/environmental variability separately. As a result, multiple small datasets can be pooled and analyzed as if they were a single large dataset, accelerating robust, high-precision diagnosis robust to gender and age variations, as well as identifying of the disease-related regions.

  • Takashi Matsubara, Koki Kusano, Tetsuo Tashiro, Ken'ya Ukai, and Kuniaki Uehara, "Deep Generative Model of Individual Variability in fMRI Images of Psychiatric Patients," IEEE Transactions on Biomedical Engineering, vol. 68, no. 2, pp. 592-605, 2021.
    IEEE
  • Koki Kusano, Tetsuo Tashiro, Takashi Matsubara, and Kuniaki Uehara, "Deep Generative State-Space Modeling of FMRI Images for Psychiatric Disorder Diagnosis," The 2019 International Joint Conference on Neural Networks (IJCNN2019), Budapest, Jul. 2019.
    Paper
  • Takashi Matsubara, Tetsuo Tashiro, and Kuniaki Uehara, "Deep Neural Generative Model of Functional MRI Images for Psychiatric Disorder Diagnosis," IEEE Transactions on Biomedical Engineering, vol. 66, no. 10, pp. 2768-2779, 2019.
    IEEE arXiv
  • Takashi Matsubara, Tetsuo Tashiro, and Kuniaki Uehara, "Structured Deep Generative Model of FMRI Signals for Mental Disorder Diagnosis," The 21st International Conference on Medical Image Computing and Computer Assisted Intervention (MICCAI2018), Granada, Sep. 2018, pp. 258-266.
    Paper
  • Tetsuo Tashiro, Takashi Matsubara, and Kuniaki Uehara, "Deep Neural Generative Model for fMRI Image Based Diagnosis of Mental Disorder," The 2017 International Symposium on Nonlinear Theory and its Applications (NOLTA2017), Cancun, Dec. 2017, pp. 700-703, 5169.

Stock Price Prediction Using Deep Generative Models of Language Information

In this study, we propose a method for daily stock price trend prediction from news articles using deep generative models. We consider the impact of each news article. First, we use a method called Paragraph Vector to represent the information in news articles as fixed-length vectors, which sufficiently captures the information in the language. Next, we represent the relationship between stock price information and language information using deep generative models and learn the parameters based on the distributed representation. By using generative models, we can represent latent variables and probabilistic processes that generate news articles, and suppress overfitting of the parameters required for that representation. We demonstrated the effectiveness of this method by performing binary classification of stock price trends for both the Japanese and American markets.

  • Takashi Matsubara, Ryo Akita, and Kuniaki Uehara, "Stock Price Prediction by Deep Neural Generative Model of News Articles," IEICE Transactions on Information and Systems, Vol.E101-D, No.4, pp.901-908, 2018.
    J-STAGE
  • Ryo Akita, Akira Yoshihara, Takashi Matsubara, and Kuniaki Uehara, "Deep Learning for Stock Prediction Using Numerical and Textual Information," the 15th IEEE/ACIS International Conference on Computer and Information Science (ICIS 2016), Okayama, Jun. 2016, pp. 945-950.
    IEEE

Image Modality Conversion for Constructing Diverse Virtual

Deep learning has been intensively investigated for autonomous driving and robotic control. These real-world problems require massive and diverse datasets, but collecting data for a wide range of conditions (such as nighttime and rainy weather) incurs significant costs. Although one approach is to develop environmental simulators to generate artificial data, discrepancies from real environments can degrade performance, and building a high-fidelity simulator itself is expensive. To overcome these challenges, this study proposes a method to augment training data by transforming real data modalities (that is, converting daytime data into nighttime data). This research was conducted as a joint research project with Toyota Central R&D Labs., Inc.

  • Shinta Masuda, Takashi Matsubara, Kuniaki Uehara, "Image Modality Conversion for Constructing Diverse Virtual Spaces," 2018 Annual Conference of the Japanese Society for Artificial Intelligence (JSAI2018) , 4M1-04, Kagoshima, June, 2018.
    Slide

Quantification of Uncertainty

Few-shot Anomaly Detection Using Grouped Data Generative Models

"Anomaly detection" is an important task in image analysis, with applications ranging from defective product inspection to medical imaging. Deep generative models enable estimation of the likelihood of high-dimensional real data, such as images, and rare samples (e.g., defective items) tend to have lower likelihood, making them detectable as anomalies. However, normal yet unseen products (e.g., newly developed products not included in the training set) also exhibit lower likelihood and are thus falsely detected as anomalies. To address this issue, we propose a deep generative model that separates features unique to each product group from those unique to individual items. Leveraging a model trained on existing products, our approach achieves "few-shot anomaly detection," thereby enabling the identification of defective items even among a small number of new product samples. This research was conducted as a joint research project with The KAITEKI Institute, Inc.

  • Kazuki Sato, Satoshi Nakata, Takashi Matsubara, and Kuniaki Uehara, "Few-shot Anomaly Detection using Deep Generative Models for Grouped Data," IEICE Transactions on Information and Systems, vol.E105-D, no.2, pp.436-440, 2022.
    J-STAGE

Anomaly Detection Using Unregularized Anomaly Score with Deep Generative Models

In general, anomaly detection refers to identifying rare instances within large datasets as "anomalies." Deep generative models, which learn to compress and reconstruct samples such as images, primarily learn typical samples and regard samples that cannot be reconstructed as anomalies. However, reconstruction failure may stem from "epistemic uncertainty" (due to insufficient training of rare samples) or "aleatoric uncertainty" (due to noise or complex shapes). Regions with high aleatoric uncertainty, such as screw holes, are frequently misdetected as anomalies despite being normal. To mitigate this problem, we decompose the likelihood in a deep generative model and uses only the component corresponding to epistemic uncertainty, termed the non-regularized anomaly score, for anomaly detection. This approach avoids being misled by visually complex regions, thereby improving detection accuracy. This research was conducted as a joint research project with AISIN AW Co., Ltd.

  • Takashi Matsubara, Kazuki Sato, Kenta Hama, Ryosuke Tachibana, and Kuniaki Uehara, "Deep Generative Model using Unregularized Score for Anomaly Detection with Heterogeneous Complexity," IEEE Transactions on Cybernetics, vol. 52, no. 6, pp. 5161-5173, 2022.
    arXiv IEEE
  • Takashi Matsubara, Ryosuke Tachibana, and Kuniaki Uehara, "Anomaly Machine Component Detection by Deep Generative Model with Unregularized Score," The 2018 International Joint Conference on Neural Networks (IJCNN2018), Rio de Janeiro, Jul. 2018, pp.4067-4074.
    IEEE

Reliability Evaluation in Image-Text Retrieval Using Bayesian Deep Learning

Evaluating the reliability of decision making of machine learning algorithms remains a major challenge. While uncertainty-based methods in Bayesian neural networks have been proposed for assessing reliability in classification and regression tasks, these methods cannot be directly applied to image-text retrieval. In this work, we define two types of uncertainty by interpreting image-text retrieval as a classification problem (posterior uncertainty) and as a regression problem (uncertainty in the embedding). Through experimentation, we found that treating image-text retrieval as a classification problem provides a more appropriate evaluation of reliability.

  • Kenta Hama, Takashi Matsubara, Kuniaki Uehara, and Jianfei Cai, "Exploring Uncertainty Measures for Image-Caption Embedding-and-Retrieval Task," ACM Transactions on Multimedia Computing, Communications, and Applications , vol. 17, no. 2, article no. 46, 2021.
    arXiv
  • Kenta Hama, Takashi Matsubara, and Kuniaki Uehara, "Image-Caption Retrieval with Evaluating Uncertainties," The 7th Japan-Korea Joint Workshop on Complex Communication Sciences (JKCCS), Pyengonchang, Jan. 2019.

Bayesian Estimation and Model Averaging of Convolutional Neural Networks by Hypernetwork

Neural networks can learn complex representations and show high performance in various tasks. However, since the data available for training is limited, they are prone to overfitting. Regularizing the training of neural networks to prevent overfitting is one of the most important challenges. In this study, we target large-scale convolutional neural networks and use hypernetworks to implicitly estimate the posterior distribution of parameters to regularize training. Additionally, since the distribution of parameters is learned, classification accuracy can be improved through model averaging.

  • Kenya Ukai, Takashi Matsubara, and Kuniaki Uehara, "Bayesian Estimation and Model Averaging of Convolutional Neural Networks by Hypernetwork," Nonlinear Theory and Its Applications, IEICE, Vol.E10-N, No.1, 2019.
    J-STAGE
  • Kenya Ukai, Takashi Matsubara, and Kuniaki Uehara, "Hypernetwork-based Implicit Posterior Estimation and Model Averaging of Convolutional Neural Networks," The 10th Asian Conference on Machine Learning (ACML2018), Beijing, Nov. 2018, pp. 176-191.
    Paper