journal article·2022-09-08Open access

A Study on the Calibrated Confidence of Text Classification Using a Variational Bayes

Juhyun Lee, Sangsung Park

Applied Sciences· Volume 12· Issue 18

Abstract

Recently, predictions based on big data have become more successful. In fact, research using images or text can make a long-imagined future come true. However, the data often contain a lot of noise, or the model does not account for the data, which increases uncertainty. Moreover, the gap between accuracy and likelihood is widening in modern predictive models. This gap may increase the uncertainty of predictions. In particular, applications such as self-driving cars and healthcare have problems that can be directly threatened by these uncertainties. Previous studies have proposed methods for reducing uncertainty in applications using images or signals. However, although studies that use natural language processing are being actively conducted, there remains insufficient discussion about uncertainty in text classification. Therefore, we propose a method that uses Variational Bayes to reduce the difference between accuracy and likelihood in text classification. This paper conducts an experiment using patent data in the field of technology management to confirm the proposed method’s practical applicability. As a result of the experiment, the calibrated confidence in the model was very small, from a minimum of 0.02 to a maximum of 0.04. Furthermore, through statistical tests, we proved that the proposed method within the significance level of 0.05 was more effective at calibrating the confidence than before.
Source: crossref · Quality: baseline

Research context

OpenAlex topics: Technology and Data Analysis · Text and Document Classification Technologies · Advanced Text Analysis Techniques

SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.

Affiliations

Institute of Engineering Research, Korea University, Seoul 02841, Korea

Department of Big Data and Statistics, Cheongju University, Chungbuk 28503, Korea

Funding

National Research Foundation of Korea (NRF)

References (74)

Show source-provided references
1. Survey on deep learning with class imbalance10.1186/s40537-019-0192-5 ↗
2. Jiawei, H., Micheline, K., and Jian, P. (2012). Data Mining: Concepts and Techniques, Elsevier. [3rd ed.].
3. Ivanović, M., and Radovanović, M. (2015). Modern Machine Learning Techniques and Their Applications. Electronics, Communications and Networks IV: Proceedings of the International Conference on Electronics, Communications and Networks, Beijing, China, 12–15 December 2014, CRC Press.10.1201/b18592-153 ↗
4. Deep learning10.1038/nature14539 ↗
5. Bojarski, M., Del Testa, D., Dworakowski, D., Firner, B., Flepp, B., Goyal, P., Jackel, L.D., Monfort, M., Muller, U., and Zhang, J. (2016). End to End Learning for Self-Driving Cars. arXiv.
6. Van den Oord, A., Dieleman, S., Zen, H., Simonyan, K., Vinyals, O., Graves, A., Kalchbrenner, N., Senior, A., and Kavukcuoglu, K. (2016). Wavenet: A Generative Model for Raw Audio. arXiv.
7. Isola, P., Zhu, J.-Y., Zhou, T., and Efros, A.A. (2017, January 21–26). Image-to-Image Translation with Conditional Adversarial Networks. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Honolulu, HI, USA.10.1109/CVPR.2017.632 ↗
8. Dermatologist-level classification of skin cancer with deep neural networks10.1038/nature21056 ↗
9. Jiang, H., Kim, B., Guan, M.Y., and Gupta, M. (2018, January 4). To Trust or Not to Trust a Classifier. Proceedings of the Advances in Neural Information Processing System, Montréal, QC, Canada.
10. A quantitative uncertainty metric controls error in neural network-driven chemical discovery10.1039/C9SC02298H ↗
11. Malinin, A. (2019). Uncertainty Estimation in Deep Learning with Application to Spoken Language Assessment. [Doctoral Dissertation, University of Cambridge].
12. A review of uncertainty quantification in deep learning: Techniques, applications and challenges10.1016/j.inffus.2021.05.008 ↗
13. Aleatory and epistemic uncertainty in probability elicitation with an example from hazardous waste management10.1016/S0951-8320(96)00077-4 ↗
14. Aleatory or epistemic? Does it matter?10.1016/j.strusafe.2008.06.020 ↗
15. Aleatoric and epistemic uncertainty in machine learning: An introduction to concepts and methods10.1007/s10994-021-05946-3 ↗
16. Kendall, A., and Gal, Y. (2018, January 24). What Uncertainties Do We Need in Bayesian Deep Learning for Computer Vision?. Proceedings of the Advances in Neural Information Processing System, Long Beach, CA, USA.
17. Probabilistic forecasts, calibration and sharpness10.1111/j.1467-9868.2007.00587.x ↗
18. Geostatistical Model Averaging for Locally Calibrated Probabilistic Quantitative Precipitation Forecasting10.1198/jasa.2011.ap10433 ↗
19. Minderer, M., Djolonga, J., Romijnders, R., Hubis, F., Zhai, X., Houlsby, N., Tran, D., and Lucic, M. (2021, January 12). Revisiting the Calibration of Modern Neural Networks. Proceedings of the Advances in Neural Information Processing System, Online.
20. Jagannatha, A., and Yu, H. (2020, January 5–10). Calibrating Structured Output Predictors for Natural Language Processing. Proceedings of the Conference Association for Computational Linguistics Meeting, Online.10.18653/v1/2020.acl-main.188 ↗
21. How Can We Know When Language Models Know? On the Calibration of Language Models for Question Answering10.1162/tacl_a_00407 ↗
22. Zhang, S., Gong, C., and Choi, E. (2021). Knowing More About Questions Can Help: Improving Calibration in Question Answering. arXiv.10.18653/v1/2021.findings-acl.172 ↗
23. Threat of Litigation and Patent Value: What Technology Managers Should Know10.5437/08956308X5602093 ↗
24. Lee, J., Kang, J., Park, S., Jang, D., and Lee, J. (2020). A Multi-Class Classification Model for Technology Evaluation. Sustainability, 12.10.3390/su12156153 ↗
25. Predicting Patent Litigation
26. Two Methodologies for Predicting Patent Litigation Outcomes: Logistic Regression Versus Classification Trees10.1111/ablj.12036 ↗
27. Sokhansanj, B.A., and Rosen, G.L. (2022). Predicting Institution Outcomes for Inter Partes Review (IPR) Proceedings at the United States Patent Trial & Appeal Board by Deep Learning of Patent Owner Preliminary Response Briefs. Appl. Sci., 12.10.3390/app12073656 ↗
28. Early detection of valuable patents using a deep learning model: Case of semiconductor industry10.1016/j.techfore.2020.120146 ↗
29. Patent Value Analysis Using Deep Learning Models—The Case of IoT Technology Mining for the Manufacturing Industry10.1109/TEM.2019.2957842 ↗
30. Technology Analysis & Strategic Management Technology Transfer in the Supply Chain Oriented to Industry 4.0: A Literature Review
31. Multimodal Deep Learning for Patent Classification
32. Adaptive Ensemble Learning Strategy Using an Assistant Classifier for Large-Scale Imbalanced Patent Categorization
33. Probabilistic Outputs for Support Vector Machines and Comparisons to Regularized Likelihood Methods
34. Guo, C., Pleiss, G., Sun, Y., and Weinberger, K.Q. (2017, January 6). On Calibration of Modern Neural Networks. Proceedings of the International Conference on Machine Learning, PMLR, Sydney, Australia.
35. Zadrozny, B., and Elkan, C. (July, January 28). Obtaining Calibrated Probability Estimates from Decision Trees and Naive Bayesian Classifiers. Proceedings of the ICML, Online.
36. Naeini, M.P., Cooper, G., and Hauskrecht, M. (2015, January 25–30). Obtaining Well Calibrated Probabilities Using Bayesian Binning. Proceedings of the Twenty-Ninth AAAI Conference on Artificial Intelligence, Austin, TX, USA.
37. Nixon, J., Dusenberry, M.W., Zhang, L., Jerfel, G., and Tran, D. (2019, January 15). Measuring Calibration in Deep Learning. Proceedings of the CVPR Workshops, Long Beach, CA, USA.
38. Thulasidasan, S., Chennupati, G., Bilmes, J., Bhattacharya, T., and Michalak, S.E. (2019, January 13). On Mixup Training: Improved Calibration and Predictive Uncertainty for Deep Neural Networks. Proceedings of the Advances in Neural Information Processing System, Vancouver, BC, Canada.10.2172/1525811 ↗
39. Zhang, L., Deng, Z., Kawaguchi, K., and Zou, J. (2022, January 17–23). When and How Mixup Improves Calibration. Proceedings of the 39th International Conference on Machine Learning, PMLR, Baltimore, MD, USA.
40. Ovadia, Y., Fertig, E., Ren, J., Nado, Z., Sculley, D., Nowozin, S., Dillon, J.V., Lakshminarayanan, B., and Snoek, J. (2019, January 13). Can You Trust Your Model’s Uncertainty? Evaluating Predictive Uncertainty Under Dataset Shift. Proceedings of the Advances in Neural Information Processing System, Vancouver, BC, Canada.
41. Chan, A.J., Alaa, A.M., Qian, Z., and van der Schaar, M. (2020, January 13–18). Unlabelled Data Improves Bayesian Uncertainty Calibration under Covariate Shift. Proceedings of the 37th International Conference on Machine Learning, PMLR, Online.
42. Hendrycks, D., and Gimpel, K. (2016). A Baseline for Detecting Misclassified and out-of-Distribution Examples in Neural Networks. arXiv.
43. Pereyra, G., Tucker, G., Chorowski, J., Kaiser, Ł., and Hinton, G. (2017). Regularizing Neural Networks by Penalizing Confident Output Distributions. arXiv.
44. Krishnan, R., and Tickoo, O. (2020, January 6). Improving Model Calibration with Accuracy versus Uncertainty Optimization. Proceedings of the Advances in Neural Information Processing System, Vancouver, BC, Canada.
45. Jiang, X., and Deng, X. (2022). Knowledge Reverse Distillation Based Confidence Calibration for Deep Neural Networks. Neural Process. Lett., 1–16.10.1007/s11063-022-10885-8 ↗
46. Xenopoulos, P., Rulff, J., Nonato, L.G., Barr, B., and Silva, C. (2022). Calibrate: Interactive Analysis of Probabilistic Model Output. arXiv.10.1109/TVCG.2022.3209489 ↗
47. A numerical simulation of the fractional order Leptospirosis model using the supervise neural network10.1016/j.aej.2022.06.013 ↗
48. Liu, Q., Wu, H., Ye, Y., Zhao, H., Liu, C., and Du, D. (2017, January 13). Patent Litigation Prediction: A Convolutional Tensor Factorization Approach. Proceedings of the International Joint Conference on Artificial Intelligence, Stockholm, Sweden.10.24963/ijcai.2018/701 ↗
49. A new ensemble method for gold mining problems: Predicting technology transfer10.1016/j.elerap.2011.06.004 ↗
50. Smarter Classification for Imbalanced Data Set and Its Application to Patent Evaluation
51. Finding the Most Efficient Technology Transfer Route Using Dijkstra Algorithm to Foster Innovation: The Case of Essential Oil Developments in the Research Center for Chemistry at the Indonesian Institute of Sciences10.14203/STIPM.2016.45 ↗
52. Sohn, K., Yan, X., and Lee, H. (2015, January 7). Learning Structured Output Representation Using Deep Conditional Generative Models. Proceedings of the Advances in Neural Information Processing System, Montreal, QC, Canada.
53. Lopez-Martin, M., Carro, B., Sanchez-Esguevillas, A., and Lloret, J. (2017). Conditional Variational Autoencoder for Prediction and Feature Recovery Applied to Intrusion Detection in IoT. Sensors, 17.10.3390/s17091967 ↗
54. Sweers, T. (2018). Autoencoding Credit Card Fraud. [Bachelor’s Thesis, Radboud University].
55. Andrei Fajardo, V., Findlay, D., Houmanfar, R., Charu, C., Jiaxi, L., and Xie, H. (2018). VOS: A Method for Variational Oversampling of Imbalanced Data Charu Jaiswal. arXiv.
56. Xu, H., Feng, Y., Chen, J., Wang, Z., Qiao, H., Chen, W., Zhao, N., Li, Z., Bu, J., and Li, Z. (2018, January 23–27). Unsupervised Anomaly Detection via Variational Auto-Encoder for Seasonal KPIs in Web Applications. Proceedings of the 2018 World Wide Web Conference, Lyon, France.10.1145/3178876.3185996 ↗
57. Lu, Y., and Xu, P. (2018). Anomaly Detection for Skin Disease Images Using Variational Autoencoder. arXiv.
58. Urteaga, I., Li, K., Shea, A., Vitzthum, V.J., Wiggins, C.H., and Elhada, N. (2021, January 6). A Generative Modeling Approach to Calibrated Predictions: A Use Case on Menstrual Cycle Length Prediction Generative Modeling for Calibrated Predictions. Proceedings of the Machine Learning for Healthcare Conference, Online.
59. Varieties of Uncertainty in Health Care: A Conceptual Taxonomy10.1177/0272989X10393976 ↗
60. Discrimination and Calibration of Clinical Prediction Models10.1001/jama.2017.12126 ↗
61. Scalable and accurate deep learning with electronic health records10.1038/s41746-018-0029-1 ↗
62. Choi, J., Moo Yi, K., Kim, J., Choo, J., Kim, B., Chang, J., Gwon, Y., and Jin Chang, H. (2021, January 20–25). VaB-AL: Incorporating Class Imbalance and Difficulty with Variational Bayes for Active Learning. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, Nashville, TN, USA.10.1109/CVPR46437.2021.00668 ↗
63. Zhu, J.-Y., Zhang, R., Pathak, D., Darrell, T., Efros, A.A., Wang, O., and Shechtman, E. (2017, January 4). Toward Multimodal Image-to-Image Translation. Proceedings of the Advances in Neural Information Processing System, Long Beach, CA, USA.
64. Variational Bayes with Intractable Likelihood10.1080/10618600.2017.1330205 ↗
65. Hwang, J.-T., Kim, B.-K., and Jeong, E.-S. (2021). Patent Value and Survival of Patents. J. Open Innov. Technol. Mark. Complex., 7.10.3390/joitmc7020119 ↗
66. Industry conditions, market share, and the firm’s ability to derive business-line profitability from diverse technological portfolios10.1016/j.jbusres.2022.05.026 ↗
67. Ren, X., Yang, T., Wang, Y., and Zeng, W. (2021). Do generative models know disentanglement? contrastive learning is all you need. arXiv.
68. Reimers, N., and Gurevych, I. (2019, January 3–7). Sentence-BERT: Sentence Embeddings using Siamese BERT-Networks. Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing and the 9th International Joint Conference on Natural Language Processing (EMNLP-IJCNLP) 2019, Hong Kong, China.10.18653/v1/D19-1410 ↗
69. An Analysis of the Scope of Copyright Protection for Application Programs10.2307/1228751 ↗
70. Macro, meso and micro perspectives of technology transfer10.1007/s10961-018-9658-4 ↗
71. University–industry collaboration: Using meta-rules to overcome barriers to knowledge transfer10.1007/s10961-018-9685-1 ↗
72. Applying data mining techniques for technology prediction in new energy vehicle: A case study in China10.1007/s11356-021-15298-z ↗
73. Two-dimensional goodness-of-fit testing in astronomy10.1093/mnras/202.3.615 ↗
74. A multidimensional version of the Kolmogorov–Smirnov test10.1093/mnras/225.1.155 ↗