Advancing Neuro-symbolic AI with Deep Knowledge-infused Learning

From Knoesis wiki
Revision as of 21:07, 27 July 2022 by Admin (Talk | contribs) (Illustration 2)

Jump to: navigation, search

After the era of Symbolic AI in the 20th century and Statistical AI in the first two decades of this century, there is a growing interest in the neuro-symbolic AI approach. It seeks to combine the respective powers and benefits of symbolic and statistical AI using knowledge graphs and deep learning. We have coined the term Knowledge-infused (deep) Learning (KiL) for a class of approaches that use a variety of knowledge at different levels of abstractions. This project will advance early and limited forms of enhancing deep learning with knowledge, called shallow and semi-deep KiL, with a more advanced form called deep-infusion. This project focuses on developing a deep learning architecture and associated algorithms that involve interleaving broader varieties of knowledge at different levels of abstractions or layers in a deep neural network.

Following activities are being pursued in this project.

  1. We are developing novel datasets that would exercise the development of novel algorithms that work in synchrony with human knowledge.
  2. Human knowledge is manifested in various forms such as rules, lexicons, relationships, relational databases, and knowledge graphs. We specifically focus on the knowledge graph's integration in deep learning algorithms (e.g., deep language models) to achieve explainability and interpretability.
  3. For explainability, specifically, user-level explainability is described as achievable through algorithms that connect its outcome with knowledge graphs and, when converged, reflect on the part of the knowledge graph that describes the predictions.
  4. For interpretability, at first, we are working towards leveraging simple and interpretable machine learning models that can help explain the internal mechanism of deep language models. Subsequently, we will work towards leveraging our understanding of the model's capability to define stratified knowledge structures like decision trees to allow the model to learn such trees at each layer with the help of a knowledge graph. With knowledge graph infusion, we provide semantic grounding to statistical models with unreasonable and non-deductive outcomes.
  5. AI systems have been stymied due to the lack of safety in data-driven AI. We have started to investigate how using domain (including process)knowledge as part of KiL methods can make AI systems safer.
  6. Building upon foundational research in this project, we have worked on several translational research opportunities that apply and advance KiL approaches. These include personalized health (specifically, mental health), personalized nutrition(specifically, management of carbohydrate intake in children with type 1 diabetes), and autonomous systems (including autonomous vehicles and smart manufacturing- see the third illustration in the attachment).

Current Research Questions in KiL

Based on our recent work under the purview of this grant, we have been asking the following questions concerning explainability, interpretability, safety, and reasonability.

  1. When do neural language models require non-parametric knowledge? We consider non-parametric knowledge as the source created/curated by humans. It includes lexicons, knowledge graphs, relational databases, etc.
  2. How do infuse non-parametric knowledge seamlessly into statistical AI models?
  3. It is known that deep neural models are learned by abstraction; how to leverage external knowledge’s inherent abstraction in enhancing the context of learned statistical representation?
  4. If the knowledge infusion is meant to happen at various layers in deep neural networks, how would the network regularize to prevent over-generalization or superfluous generations?
  5. We have established that the attention matrix in current transformer models makes the model reactive to global and local information in the input. It does by token-by-token square matrix. If we want to perform infusion, we must introduce two new matrices: (a) token and entity and (b) entity and entity. Simple matrix multiplication won’t work as these are out of the distribution matrix. Hence, we need to seek ways of creating a knowledge-aware attention matrix for the model from (a)token-by-token matrix, (b) token and entity, and (c) entity and entity matrix.
  6. Layered knowledge infusion might result in high-energy nodes contributing to the outcome. This is counter to the current softmax prediction. How to pick the most probable outcome? This would require us to explore marginalized loss functions using infused knowledge and input.
  7. How do you enable the generation of user-level explanations? (see the second illustration in the attachment).
  8. How to enforce safety constraints in model generations. This has been a pressing need since models tend to generate risky sentences (see the fourth illustration in the attachment).

Significant Results So Far

An overview of significant results reported in the publications and other dissemination is provided below.

  1. We explored various statistical bottlenecks in deep neural language models from the perspective of user-level explainability, interpretability, and safety. These pre-trained models are efficient, but the datasets they are trained on are not grounded in knowledge. Sheth et al. and Gupta et al. found that models hallucinate while generating responses leading to a factually incorrect or superfluous response. We investigated various methods of controlling this hallucination.
  2. We hypothesize that humans communicate through a contextual process of understanding and response. This process can either be an n-ary tree, a flat graph, or any structure of the conceptual flow. But, we face a challenge in terms of datasets. So, we created datasets cycling through a month-long annotation, evaluation, and quality check process. These datasets have been constructed under the purview of this grant and will be made available. The procedure we laid out in constructing these datasets contributes to the significance of our results. Every dataset was created automatically following deep learning and clinical knowledge. Subject Matter Experts were tasked to evaluate our labeling process. By this means, we not only checked that our knowledge-infused learning pipeline was accurate but also scaled the annotation-evaluation process by multiple folds and reduced time. Roy et al. could gather ten Subject Matter Experts' evaluations on five sets of outcomes from 5 different deep language models.
  3. We found that the simple neural language model can provide explainable results, and we also saw that experts achieved satisfactory agreement scores of >75% with simple language models.
  4. Further, we could achieve task transferability in models as they are trained on relatable clinical process knowledge-driven datasets. So far, we have found convincing results in Depression, Anxiety, and Suicide-related research. This also marks our first concrete step in realizing Knowledge-infused Learning.
  5. Some challenges arise when mental health conditions are comorbid with diseases like Cardiovascular, where we require contextual information on diseases and gender along with users' expressions.
  6. Next, we explored the domain of conversational AI as chatbots need to be safe when they communicate with the user with depression, anxiety, or suicidal tendencies (Gaur et al. CIKM). We have developed novel evaluation metrics and interpretable and explainable algorithms for process knowledge infusion in the Knowledge-infused Learning paradigm (Roy et al. ACL-IJCNLP).
  7. We studied mental health conversations related to Cardiovascular disease on social media, which requires domain knowledge. We developed knowledge-assisted masked language models in a task adaptive multi-task learning paradigm. We could differentiate the gender language and gender-specific symptoms based on user posts and comments. Lokala et al. proposed framework GeM fall under shallow knowledge-infused learning as we use external lexicons on Anxiety, Depression, and Gender for Knowledge-aware Entity Masking.
  8. The diverse forms of knowledge we infused into statistical AI are correlational. They are defined by word co-occurrence, synonymy linkage, and others but aren't causal. Representation of causality in AI systems using knowledge graphs can further improve explainability. Jamini and Sheth proposed a neat architecture on why causal knowledge graphs (CKGs) are needed, what modifications need to be made in existing knowledge graphs, and how infusion would occur.
  9. Within the scope of Knowledge-infused Learning, the causality aspect made us explore the autonomous driving domain. There are various scenarios in autonomous driving where the vehicle needs to decide based on what it has learned in other similar situations. Situations are scenes, and every scene has an interconnected set of entities that describe the scene. Wickramarachchi et al. developed a scene ontology for autonomous driving use cases and used it to extract entities from scene descriptions. An interconnection of scene entities is what is termed a scene graph. Such a graph improves machine perception in autonomous vehicles and can define sensible actions. Scene graphs and actions are absorbed by the architecture proposed by Jamini et al. to construct CausalKG.
  10. Along with mental health and healthcare in general, we are exploring the utility of knowledge-infused learning in autonomous driving. We find synchrony between these domains as the machine is tasked to provide action. Thus a correlation-alone knowledge graph falls short in expressing high-order semantic knowledge as expressed by humans.
  11. In a complementary direction, we studied the COVID-19 pandemic with a motive to help policymakers with explainable AI tools. Sivaraman et al. presented EXO-SIR, an epidemiological model supported by a component that considers external knowledge from textual reports to bootstrap SIR in estimating the likelihood of a rise in infections.

Illustrations: Key Concepts and Advances

Illustration 1

Illustration 01 - Progress in this project includes development of a deep knowledge infusion architecture, showing use of stratified knowledge (i.e., knowledge at different levels of abstractions) to modulate parameters at different layers in the deep neural network architecture. This figure shows adaptation of this advance in this project to the use case in autism at the center of the NSF AI Institute proposal submitted in May 2022 involving 11 institutions (Prof. Sheth is also the PI on that pending proposal)

Illustration 2

Illustration 02 - This figure shows the application of user-level explainability enabled by use of knowledge graph in KiL in the use case of autism to explain the reasons of referring a student to a school psychologist.

Illustration 3

File:Illustration03.jpg
Figure 3. Reddit is a rich source for bringing crowd perspective in training DLMs over conversational data. On the left is a sample post from r/depression help which sees inquisitive interaction from other Reddit users. At the top-right are the Follow-up Questions (FQs) asked by the Reddit users in the comments. These FQs are aimed at understanding the severity of the mental health situation of the user and are hence, diagnostically relevant. At the bottom-right are the questions generated by DLMs. It can be seen that these are not suitable FQs.

Illustration 4

File:Illustration04.jpg
Figure 4. The failure of large language models in supporting safety has been well publicized (left). The figure on the right shows the use of domain knowledge as part of Knowledge-infused Learning to ensure the conversation is medically valid and safe (as discussed in one of our publications).

Dissemination

Publications

  1. Wickramarachchi, Ruwan and Henson, Cory and Sheth, Amit. (2022). Knowledge-Based Entity Prediction for Improved Machine Perception in Autonomous Systems. IEEE Intelligent Systems. 37 (4), pp. 1 to 7. doi:https://doi.org/10.1109/MIS.2022.3181015 Fulltext Citation details
  2. Lokala, Usha and Srivastava, Aseem and Dastidar, Triyasha and Chakraborty, Tanmoy and Akhtar, Md. Shad andPanahiazar, Maryam and Sheth, Amit. (2022). A Computational Approach to Understand Mental Health from Reddit:Knowledge-Aware Multitask Learning Framework. Sixteenth International AAAI Conference on Web and Social Media (ICWSM). 2022, pp. 640-650.Fulltext Citation details
  3. Sivaraman, Nirmal Kumar and Gaur, Manas and Baijal, Shivansh and Muthiah, Sakthi Balan and Sheth, Amit. (2022). Exo-SIR: an epidemiological model to analyze the impact of exogenous spread of infection. International Journal of Data Science and Analytics. doi:https://doi.org/10.1007/s41060-022-00334-z ; Fulltext Citation details
  4. Jaimini, Utkarshani and Sheth, Amit. CausalKG: Causal Knowledge Graph Explainability Using Interventional andCounterfactual Reasoning. IEEE Internet Computing. 26 (1), 2022, pp: 43 - 50. doi:https://doi.org/10.1109/MIC.2021.3133551Fulltext Citation details
  5. Gupta, Shrey; Agarwal, Anmol; Gaur, Manas; Roy, Kaushik; Narayanan, Vignesh; Kumaraguru, Ponnurangam; Sheth, Amit. "Learning to Automate Follow-up Question Generation using Process Knowledge for Depression Triage on Reddit Posts," The Eighth Workshop on Computational Linguistics and Clinical Psychology: Mental Health in the Face of Change (CLPsych20022 @ NAACL2022), 15 July 2022.
  6. Venkataramanan, Revathy; Padhee, Swati; Rao, Saini Rohan; Sundarajan, Anirudh; Kaoshik, Ronak; and Amit Sheth. “Ki-Cook: Knowledge infused Multimodal Cooking Representation Learning”. Submitted for publication.
  7. Gaur, Manas, Kaushik Roy, Anmol Agarwal, Shrey Gupta, Amit Sheth, Ponnurangam Kumaragaru, "Research PK-iL: Explainable Dataset and Method for Suicidality Detection on Reddit using Process Knowledge in C-SSRS", Under Review.
  8. Gaur, Manas, Gunaratna, Kalpa, Bhatt, Shreyansh, Sheth, Amit, Special Issue on Knowledge Infused Learning in IEEE Internet Computing (Link: https://www.computer.org/digital-library/magazines/ic/call-for-papers-special-issue-on-knowledge-infused-learning) (In print)

Dataset

  • PRIMATE is a mental health dataset containing 2003 user posts from Reddit with each post having 3 attributes: 1) title, 2) post text, and 3) binary yes/no labels for each of the 9 questions in PHQ-9 depending on whether the question is answerable using the content present in the post text & title. The dataset can be used to guide language models to generate non-redundant questions which have not already been answered in the user post. The dataset is released at: https://github.com/primate-mh/Primate2022

Keynotes and Invited Talks

  1. Sheth, A., Don’t Handicap AI without Explicit Knowledge, Keynotes at the IEEE Services, Sept 2021, and the DEXA2021, Oct 2021.
  2. Sheth, A., Knowledge-infused NLU for Addiction and Mental Health Research, Keynotes at the MAISoN21, Aug 2021, and the ASONAM21, Sept 2021.
  3. Sheth, A., Augmented Personalized Health: an explicit knowledge enhanced neural symbolic Health approach to patient empowerment for managing chronic disease burden. Keynote at the SWAT4HCLS (Semantic Web Applications and Tools for Healthcare and Life Sciences), ​​January 2022.

Tutorials

  1. Roy, Kaushik; Gaur, Manas, Zhang, Qi; Sheth, A. Knowledge-infused Reinforcement Learning, the Knowledge Graph Conference (KGC 20220), May 2022. https://aiisc.ai/kirl/
  2. Gaur, Manas and Roy, Kaushik. (2022). Tutorial: Knowledge Infused Reinforcement Learning for Social Good Applications. International Semantic Intelligence Conference, May 2020. 17-19.

Prior Relevant Publications on KiL and related topics

  1. Gaur, M., Kursuncu, U., Alambo, A., Sheth, A., Daniulaityte, R., Thirunarayan, K., & Pathak, J. (2018). “Let Me Tell You About Your Mental Health!”: Contextualized Classification of Reddit Posts to DSM-5 for Web-based Intervention. Proceedings of the 27th ACM International Conference on Information and Knowledge Management, 753–762. https://doi.org/10.1145/3269206.3271732
  2. Sheth, A., Yip, H. Y., & Shekarpour, S. (2019). Extending Patient-Chatbot Experience with Internet-of-Things and Background Knowledge: Case Studies with Healthcare Applications. IEEE Intelligent Systems, 34(4), 24–30. https://doi.org/10.1109/MIS.2019.2905748
  3. Kadariya, D., Venkataramanan, R., Yip, H. Y., Kalra, M., Thirunarayanan, K., & Sheth, A. (2019). kBot: Knowledge-Enabled Personalized Chatbot for Asthma Self-Management. 2019 IEEE International Conference on Smart Computing (SMARTCOMP), 138–143. https://doi.org/10.1109/SMARTCOMP.2019.00043
  4. Gaur, M., Alambo, A., Sain, J. P., Kursuncu, U., Thirunarayan, K., Kavuluru, R., … Pathak, J. (2019). Knowledge-aware Assessment of Severity of Suicide Risk for Early Intervention. The World Wide Web Conference, 514–525.https://doi.org/10.1145/3308558.3313698
  5. Bhatt, S., Padhee, S., Sheth, A., Chen, K., Shalin, V., Doran, D., & Minnery, B. (2019). Knowledge Graph Enhanced Community Detection and Characterization. Proceedings of the Twelfth ACM International Conference on Web Search and Data Mining, 51–59. https://doi.org/10.1145/3289600.3291031
  6. Alambo, A., Gaur, M., Lokala, U., Kursuncu, U., Thirunarayan, K., Gyrard, A., Sheth, A., Welton, R. S., Pathak, J. (2019). Question Answering for Suicide Risk Assessment Using Reddit. 2019 IEEE 13th International Conference on Semantic Computing (ICSC), 468-473. Retrieved from https://doi.org/10.1109/ICOSC.2019.8665525
  7. Yadav, S., Pallagani, V., & Sheth, A. (2020). Medical Knowledge-enriched Textual Entailment Framework. ArXiv:2011.05257 [Cs]. http://arxiv.org/abs/2011.05257
  8. Wickramarachchi, R., Henson, C., & Sheth, A. (2020). An evaluation of knowledge graph embeddings for autonomous driving data: Experience and practice. Proceedings of the AAAI 2020 Spring Symposium on Combining Machine Learning and Knowledge Engineering in Practice (AAAI-MAKE 2020)
  9. Purohit, H., Shalin, V. L., Sheth, A. P., & Sheth, A. (2020). Knowledge graphs to empower humanity-inspired AI systems. IEEE Internet Computing, 24(4), 48–54. https://doi.org/10.1109/MIC.2020.3013683
  10. Kursuncu, U., Gaur, M., & Sheth, A. (2020). Knowledge infused learning (K-IL): Towards deep incorporation of knowledge in deep learning. Proceedings of the AAAI 2020 Spring Symposium on Combining Machine Learning and Knowledge Engineering in Practice (AAAI-MAKE).
  11. Gaur, M., Desai, A., Faldu, Keyur, & Sheth, A. (2020). Explainable AI using knowledge graphs. ACM CoDS-COMAD Conference. https://aiisc.ai/xaikg/
  12. Yadav, S., Lokala, U., Daniulaityte, R., Thirunarayan, K., Lamy, F., & Sheth, A. (2021). “When they say weed causes depression, but it’s your fav antidepressant”: Knowledge-aware attention framework for relationship extraction. PLOS ONE, 16(3), e0248299. https://doi.org/10.1371/journal.pone.0248299
  13. Xia, K., Saidy, C., Kirkpatrick, M., Anumbe, N., Sheth, A., & Harik, R. (2021). Towards semantic integration of machine vision systems to aid manufacturing event understanding. Sensors, 21(13), 4276. https://doi.org/10.3390/s21134276
  14. Wickramarachchi, R., Henson, C., & Sheth, A. (2021). Knowledge-infused Learning for Entity Prediction in Driving Scenes. Frontiers in Big Data, 4.https://doi.org/10.3389/fdata.2021.759110
  15. Sheth, A., & Thirunarayan, K. (2021). The duality of data and knowledge across the three waves of AI. IT Professional, 23(3), 35–45. https://doi.org/10.1109/MITP.2021.3070985
  16. Sheth, A., Shalin, V. L., & Kursuncu, U. (2021, April). Defining and detecting toxicity on social media: Context and knowledge are key. Neurocomputing. https://doi.org/10.1016/j.neucom.2021.11.095
  17. Roy, K., Zhang, Q., Gaur, M., & Sheth, A. (2021). Knowledge infused policy gradients for adaptive pandemic control. Proceedings of the AAAI 2021 Spring Symposium on Combining Machine Learning and Knowledge Engineering (AAAI-MAKE 2021).
  18. Roy, K., Zhang, Q., Gaur, M., & Sheth, A. (2021). Knowledge infused policy gradients with upper confidence bound for relational bandits, In Joint European Conference on Machine Learning and Knowledge Discovery in Databases, pp. 35-50. Springer, Cham, 2021..
  19. Roy, K., Lokala, U., Khandelwal, V., & Sheth, A. (2021). “Is depression related to cannabis?”: A knowledge-infused model for entity and relation extraction with limited Supervision. Proceedings of the AAAI 2021 Spring Symposium on Combining Machine Learning and Knowledge Engineering (AAAI-MAKE 2021).
  20. Jaimini, U., & Sheth, A. (2021). Personalized digital phenotype score, healthcare management and intervention strategies using Knowledge enabled Digital Health Framework for pediatric asthma. In Asthma. IntechOpen. https://doi.org/10.5772/intechopen.97430
  21. Gaur, M., Roy, K., Sharma, A., Srivastava, B., & Sheth, A. (2021, August). “Who can help me?”: Knowledge Infused Matching of Support Seekers and Support Providers during COVID-19 on Reddit. In 2021 IEEE 9th International Conference on Healthcare Informatics (ICHI) (pp. 265-269). IEEE.
  22. Gaur, M., Faldu, K., & Sheth, A. (2021). Semantics of the black-box: Can knowledge graphs help make deep learning systems more interpretable and explainable? IEEE Internet Computing, 25 (1), Jan-Feb 2021, pp. 51–59. https://doi.org/10.1109/MIC.2020.3031769
  23. Gaur, M., Aribandi, V., Alambo, A., Kursuncu, U., Thirunarayan, K., Beich, J., Pathak, J., & Sheth, A. (2021). Characterization of time-variant and time-invariant assessment of suicidality on Reddit using C-SSRS. PloS one, 16(5), e0250448. https://doi.org/10.1371/journal.pone.0250448
  24. Dolbir, N., Dastidar, T., & Roy, K. (2021). NLP is not enough - Contextualization of user input in chatbots. KGC Workshop on Knowledge-infused Learning. https://aiisc.ai/KiL2021/papers/KiL2021.pdf
  25. Sawhney, R., Neerkaje, A., & Gaur, M. (2022, May). A Risk-Averse Mechanism for Suicidality Assessment on Social Media. In Proceedings of the 60th Annual Meeting of the Association for Computational Linguistics (Volume 2: Short Papers) (pp. 628-635).
  26. Lokala, U., Srivastava, A., Dastidar, T. G., Chakraborty, T., Akthar, M. S., Panahiazar, M., & Sheth, A. (2022). A Computational Approach to Understand Mental Health from Reddit: Knowledge-aware Multitask Learning Framework. Proceedings of the Sixteenth International AAAI Conference on Web and Social Media (ICWSM2022)640 ArXiv:2203.11856 [Cs]. http://arxiv.org/abs/2203.11856
  27. Gaur, M., Gunaratna, K., Sriniasan, V., & Jin, H. (2022). ISEEQ: Information seeking question generation using dynamic meta-information retrieval and knowledge graphs. 36th AAAI Conference.
  28. Wickramarachchi, R., Henson, C., & Sheth, A. (2022). Knowledge-based Entity Prediction for Improved Machine Perception in Autonomous Systems, arXiv:2203.16616 [Cs]. https://arxiv.org/abs/2203.16616
  29. U. Jaimini and A. Sheth, "CausalKG: Causal Knowledge Graph Explainability Using Interventional and Counterfactual Reasoning," in IEEE Internet Computing, vol. 26, no. 1, pp. 43-50, 1 Jan.-Feb. 2022, doi: 10.1109/MIC.2021.3133551. [Ack: EAGER]
  30. Revathy Venkataramanan, Swati Padhee, Saini Rohan Rao, Anirudh Sundarajan, Ronak Kaoshik, Amit Sheth. “Ki-Cook: Knowledge infused Multimodal Cooking Representation Learning”. Submitted for publication.

Tutorials

  1. Gaur, Manas; Kursuncu, Ugur; Sheth, Amit; Wickramarachchi, Ruwan; and Yadav, Shweta. "ACM HT 2020 Tutorial: Knowledge-infused Deep Learning." In Proceedings of the 31st ACM Conference on Hypertext and Social Media, pp. 309-310. 2020. http://kidl2020.aiisc.ai/
  2. Gaur, M., Desai, A., Faldu, Keyur, & Sheth, A. Explainable AI using Knowledge Graphs. ACM CoDS-COMAD Conference, 2020. https://aiisc.ai/xaikg/
  3. Gaur M., Roy, Kaushik, Zhang, Qi, Sheth Amit; Tutorial on Knowledge-infused Reinforcement Learning at Knowledge Graph Conference, Link: https://www.knowledgegraph.tech/kgc-2022-tutorial-knowledge-infused-reinforcement-learning/
  4. Gaur M., Roy, Kaushik, Sheth, Amit; Tutorial on Knowledge-infused Reinforcement Learning at International Semantic Intelligence Conference (Link:

Conceptual basis that led to KiL: (conceptually related but not specifically KiL)

  1. A. Sheth, S. Perera, S. Wijeratne, and K. Thirunarayan. 2017. Knowledge will propel machine understanding of content: extrapolating from current examples. In Proceedings of the International Conference on Web Intelligence (WI '17). ACM, New York, NY, USA, 1-9. DOI: https://doi.org/10.1145/3106426.3109448 Keynote: 2017
  2. A. Sheth, K. Thirunarayanan, The Duality of Data and Knowledge Across the Three Waves of AI, IT Professional (IEEE’s 75th anniversary special issue), May 2021.


Funding

  • NSF Award#: 2133842
  • EAGER: Advancing Neuro-symbolic AI with Deep Knowledge-infused Learning
  • Timeline: 01 July 2021 - 30 June 2022
  • Award Amount: $139,999


Personnel