Data augmentation in microscopic images for material data mining
Source abstract
Abstract
Abstract Recent progress in material data mining has been driven by high-capacity models trained on large datasets. However, collecting experimental data (real data) has been extremely costly owing to the amount of human effort and expertise required. Here, we develop a novel transfer learning strategy to address problems of small or insufficient data. This strategy realizes the fusion of real and simulated data and the augmentation of training data in a data mining procedure. For a specific task of grain instance image segmentation, this strategy aims to generate synthetic data by fusing the images obtained from simulating the physical mechanism of grain formation and the “image style” information in real images. The results show that the model trained with the acquired synthetic data and only 35% of the real data can already achieve competitive segmentation performance of a model trained on all of the real data. Because the time required to perform grain simulation and to generate synthetic data are almost negligible as compared to the effort for obtaining real data, our proposed strategy is able to exploit the strong prediction power of deep learning without significantly increasing the experimental burden of training data preparation.
Source: crossref · Quality: baseline
Research context
OpenAlex topics: Image Processing Techniques and Applications · Mineral Processing and Grinding · Cell Image Analysis Techniques
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
References (46)
Show source-provided references
1. Lookman, T., Alexander, F. J. & Rajan, K. Information Science for Materials Discovery and Design. (Springer Cham. Press, Switzerland, 2016).10.1007/978-3-319-23871-5 ↗
2. Butler, K. T., Davies, D. W., Cartwright, H., Isayev, O. & Walsh, A. Machine learning for molecular and materials science. Nature 559, 547–555 (2018).10.1038/s41586-018-0337-2 ↗
3. Raccuglia, P. et al. Machine-learning-assisted materials discovery using failed experiments. Nature 533, 73–76 (2016).10.1038/nature17439 ↗
4. Luna, P. D., Wei, J., Bengio, Y., Aspuru-Guzik, A. & Sargent, E. Use machine learning to find energy materials. Nature 552, 23–27 (2017).10.1038/d41586-017-07820-6 ↗
5. Sanchez-Lengeling, B. & Aspuru-Guzik, A. Inverse molecular design using machine learning: Generative models for matter engineering. Science 361, 360–365 (2018).10.1126/science.aat2663 ↗
6. Wu, S. et al. Machine-learning-assisted discovery of polymers with high thermal conductivity using a molecular design algorithm. npj Comput. Mater. 5, 66 (2019).10.1038/s41524-019-0203-2 ↗
7. Wang, C. S., Fu, H. D., Jiang, L., Xue, D. Z. & Xie, J. X. A property-oriented design strategy for high performance copper alloys via machine learning. npj Comput. Mater. 5, 87 (2019).10.1038/s41524-019-0227-7 ↗
8. Wu, Y. J., Fang, L. & Xu, Y. B. Predicting interfacial thermal resistance by machine learning. npj Comput. Mater. 5, 56 (2019).10.1038/s41524-019-0193-0 ↗
9. Dursun, T. & Soutis, C. Recent developments in advanced aircraft aluminium alloys. Mater. Des. 56, 862–871 (2014).10.1016/j.matdes.2013.12.002 ↗
10. Hu, J., Shi, Y. N., Sauvage, X., Sha, G. & Lu, K. Grain boundary stability governs hardening and softening in extremely fine nanograined metals. Science 355, 1292–1296 (2017).10.1126/science.aal5166 ↗
11. Sonka, M., Hlavac, V. & Boyle, R. Image Processing, Analysis, and Machine Vision. 4th edn (Cengage Learning. Press, Boston, 2014).
12. Krizhevsky, A., Sutskever, I. & Hinton, G. E. Imagenet classification with deep convolutional neural networks. NeurIPS. 1097–1105 (2012).
13. Long, J., Shelhamer, E. & Darrell, T. Fully convolutional networks for semantic segmentation. CVPR. 3431–3440 (2015).10.1109/CVPR.2015.7298965 ↗
14. Lewis, A. C. & Howe, D. Future directions in 3D materials science: outlook from the first international conference on 3D materials science. JOM 66, 670–673 (2014).10.1007/s11837-014-0883-5 ↗
15. Li, W., Field, K. G. & Morgan, D. Automated defect analysis in electron microscopic images. npj Comput Mater. 4, 35 (2018).10.1038/s41524-018-0094-7 ↗
16. Azimi, S. M., Britz, D., Engstler, M., Fritz, M. & Mucklich, F. Advanced steel microstructural classification by deep learning methods. Sci. Rep. 8, 2128 (2018).10.1038/s41598-018-20037-5 ↗
17. Ma, B. Y. et al. Deep learning-based image segmentation for al-la alloy microscopic images. Symmetry 10, 107 (2018).10.3390/sym10040107 ↗
18. Richter, S. R., Vineet, V., Roth, S. & Koltun, V. Playing for data: ground truth from computer games. ECCV. 102–118 (2016).10.1007/978-3-319-46475-6_7 ↗
19. Anderson, M. P., Grest, G. S. & Srolovitz, D. J. Computer simulation of normal grain growth in three dimensions. Philos. Mag. B 59, 293–329 (1989).10.1080/13642818908220181 ↗
20. Radhakrishnan, B. & Zacharia, T. Simulation of curvature-driven grain growth by using a modified Monte Carlo algorithm. Metall. Mater. Trans. A. 26, 167–180 (1995).10.1007/BF02669802 ↗
21. Gatys, L. A., Ecker, A. S. & Bethge, M. Image style transfer using convolutional neural networks. CVPR. 2414–2423 (2016).10.1109/CVPR.2016.265 ↗
22. Gatys, L. A., Ecker, A. S. & Bethge, M. A neural algorithm of artistic style. Preprint at https://arxiv.org/abs/1508.06576 (2015).
23. Isola, P., Zhu, J. Y., Zhou, T. & Efros, A. A. Image-to-image translation with conditional adversarial networks. CVPR. 1125–1134 (2017).10.1109/CVPR.2017.632 ↗
24. Zhu, J. Y., Park, T., Isola, P. & Efros, A. A. Unpaired image-to-image translation using cycle-consistent adversarial networks. ICCV. 2223–2232 (2017).10.1109/ICCV.2017.244 ↗
25. Shrivastava, A. et al. Learning from simulated and unsupervised images through adversarial training. CVPR. 2107–2116 (2017).10.1109/CVPR.2017.241 ↗
26. Choi, Y. et al. Stargan: unified generative adversarial networks for multi-domain image-to-image translation. CVPR. 8789–8797 (2018).10.1109/CVPR.2018.00916 ↗
27. Zhu, J. Y. et al. Toward multimodal image-to-image translation. NeurIPS. 465–476 (2017).
28. Ronneberger, O., Fischer, P. & Brox, T. U-net: convolutional networks for biomedical image segmentation. MICCAI. 234–241 (2015).10.1007/978-3-319-24574-4_28 ↗
29. Falk, T. et al. U-Net: deep learning for cell counting, detection, and morphometry. Nat. Methods 16, 67–70 (2019).10.1038/s41592-018-0261-2 ↗
30. Kaggle data science bowl 2018. https://www.kaggle.com/c/data-science-bowl-2018/overview/evaluation (2019).
31. Lin, T. Y. et al. Microsoft COCO: common objects in context. ECCV. 740–755 (2014).10.1007/978-3-319-10602-1_48 ↗
32. Rand, W. M. Objective criteria for the evaluation of clustering methods. J. Am. Stat. Assoc. 66, 846–850 (1971).10.1080/01621459.1971.10482356 ↗
33. Hubert, L. & Arabie, P. Comparing partitions. J. Classif. 2, 193–218 (1985).10.1007/BF01908075 ↗
34. Vinh, N. X., Epps, J. & Bailey, J. Information theoretic measures for clusterings comparison: Variants, properties, normalization and correction for chance. JMLR 11, 2837–2854 (2010).
35. Python language reference. http://www.python.org (2019).
36. Pytorch. https://pytorch.org/ (2019).
37. Laganière, R. OpenCV 3 Computer Vision Application Programming Cookbook. 3rd edn (Packt Publishing Ltd. Press, Birmingham, 2017).
38. Otsu, N. A threshold selection method from gray-level histograms. IEEE T Syst. Man CY-S 9, 62–66 (1979).10.1109/TSMC.1979.4310076 ↗
39. Canny, J. A computational approach to edge detection. TPAMI 6, 679–698 (1986).10.1109/TPAMI.1986.4767851 ↗
40. Roerdink, J. B. & Meijster, A. The watershed transform: definitions, algorithms and parallelization strategies. Fund Inform. 41, 187–228 (2000).
41. Jain, A. K. Data clustering: 50 years beyond K-means. ICPR 31, 651–666 (2010).
42. Iglovikov, V. & Shvets, A. Ternausnet: U-net with vgg11 encoder pre-trained on imagenet for image segmentation. Preprint at https://arxiv.org/abs/1801.05746 (2018).
43. Yamada, H. et al. Predicting materials properties with little data using shotgun transfer learning. ACS Central Sci. 5, 1717–1730 (2019).10.1021/acscentsci.9b00804 ↗
44. Goodfellow, I. J. et al. Generative adversarial. Netw. NeurIPS 3, 2672–2680 (2014).
45. Pix2Pix in Pytorch. https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix (2019).
46. Li, C. & Wand, M. Precomputed real-time texture synthesis with markovian generative adversarial networks. ECCV. 702–716 (2016).10.1007/978-3-319-46487-9_43 ↗