theme_to_description

app.py

@@ -61,6 +61,9 @@ def load_assets():
     base_path = Path(__file__).parent
     with open(base_path/"label_to_theme.json") as f:
         label_to_theme = json.load(f)
+    
+    with open(base_path/"theme_to_descripiton.json") as f:
+        theme_to_description = json.load(f)
 
     class_weights = torch.load(f"{base_path}/class_weights.pth").to(device)
 
@@ -69,7 +72,7 @@ def load_assets():
     model.eval()
 
     tokenizer = DebertaV2Tokenizer.from_pretrained(MODEL_NAME)
-    return model, tokenizer, label_to_theme, device
+    return model, tokenizer, device, label_to_theme, theme_to_description
 
 
 def preprocess_text(text, tokenizer, max_length=MAX_LENGTH):
@@ -115,7 +118,7 @@ def main():
     st.title("Paper Classification App")
     st.write("Classify research papers using DeBERTa model")
 
-    model, tokenizer, label_to_theme, device = load_assets()
+    model, tokenizer, device, label_to_theme, theme_to_description = load_assets()
 
     title = st.text_input("Title")
     abstract = st.text_area("Abstract")
@@ -136,13 +139,13 @@ def main():
             probabilities = predict(text, model, tokenizer, device)
             themes = get_themes(probabilities, label_to_theme)
         
-        st.success("Classification results:")
+        st.success("Predicted themes (click to expand):")
         # for theme, prob in themes:
         #     st.write(f"- {theme}: {prob:.2%}")
 
         for theme, prob in themes:
             with st.expander(f"{theme} ({prob:.1%})"):
-                st.markdown(f"**Description**: ")
+                st.markdown(f"**Description**: {theme_to_description[theme]}")
 
 
 if __name__ == "__main__":

theme_to_descripiton.json

@@ -0,0 +1,49 @@
+{
+    "cs.AI": "Artificial Intelligence: Focuses on creating intelligent systems capable of reasoning, learning, and problem-solving through techniques like machine learning and knowledge representation.",
+    "physics.soc-ph": "Physics and Society: Applies physics methodologies to social systems, analyzing phenomena like opinion dynamics, urban growth, and network behavior using statistical physics approaches.",
+    "stat.ML": "Machine Learning (Statistics): Develops statistical frameworks and algorithms for pattern recognition, predictive modeling, and data-driven decision making.",
+    "cs.CE": "Computational Engineering: Uses numerical methods and algorithms to solve complex problems in engineering design, financial modeling, and scientific simulations.",
+    "cs.DB": "Database Systems: Studies the design, implementation, and optimization of systems for efficient data storage, retrieval, and management.",
+    "cs.CL": "Natural Language Processing: Focuses on computational understanding and generation of human language, including machine translation and text analysis.",
+    "cs.NA": "Numerical Analysis: Develops algorithms for solving mathematical problems numerically, with emphasis on stability, accuracy, and computational efficiency.",
+    "cs.CY": "Computers and Society: Examines ethical implications, digital privacy, and societal impacts of computing technologies.",
+    "cs.GT": "Game Theory: Analyzes strategic decision-making in competitive situations through mathematical models of conflict and cooperation.",
+    "cs.SI": "Social Networks: Investigates structure and dynamics of online communities, information diffusion, and network-based prediction models.",
+    "stat.AP": "Applied Statistics: Implements statistical methods to solve practical problems in fields like medicine, economics, and environmental science.",
+    "cs.DL": "Deep Learning: Specializes in neural networks with multiple processing layers for complex pattern recognition in data-rich environments.",
+    "math.ST": "Statistical Theory: Develops fundamental principles of statistical inference, probability theory, and hypothesis testing.",
+    "nlin.AO": "Complex Systems: Studies emergent behavior in nonlinear systems, including biological networks and self-organizing phenomena.",
+    "cs.LO": "Formal Logic: Applies mathematical logic to computer science problems including program verification and automated theorem proving.",
+    "cs.MM": "Multimedia Computing: Handles processing and analysis of images, video, audio, and other rich media formats.",
+    "cond-mat.dis-nn": "Disordered Systems: Investigates materials with irregular structures and their computational analogs in neural network models.",
+    "cs.DM": "Discrete Mathematics: Focuses on mathematical structures like graphs and combinatorics with computer science applications.",
+    "cs.CC": "Computational Complexity: Classifies problems by their inherent difficulty and resource requirements for computation.",
+    "stat.CO": "Statistical Computing: Develops numerical algorithms and computational methods for statistical analysis.",
+    "cs.DC": "Distributed Computing: Coordinates computation across multiple networked devices for scalable problem solving.",
+    "cs.IT": "Information Theory: Quantifies information transmission and processing, including error-correcting codes and data compression.",
+    "cs.DS": "Data Structures: Designs efficient methods for organizing and manipulating data collections in computer systems.",
+    "cs.SY": "Control Systems: Develops theory and applications for automated control of dynamic systems in engineering and robotics.",
+    "q-bio.QM": "Quantitative Biology: Applies mathematical modeling and computational methods to biological systems analysis.",
+    "cs.PL": "Programming Languages: Creates and analyzes language constructs for efficient and secure software development.",
+    "cs.RO": "Robotics: Integrates sensing, actuation, and control algorithms for autonomous machine operation.",
+    "cs.NE": "Neurocomputing: Develops brain-inspired algorithms including neural networks and evolutionary optimization methods.",
+    "cs.CR": "Cybersecurity: Protects information systems through cryptography, secure protocols, and vulnerability analysis.",
+    "cs.MA": "Multi-Agent Systems: Coordinates intelligent agents in distributed environments for collaborative problem solving.",
+    "q-bio.NC": "Neural Cognition: Studies brain mechanisms underlying perception, learning, and decision-making processes.",
+    "cs.LG": "Machine Learning: Creates algorithms that automatically improve through experience using statistical techniques.",
+    "cs.GR": "Computer Graphics: Generates and manipulates visual content through rendering techniques and geometric modeling.",
+    "physics.data-an": "Physics Data Analysis: Develops specialized methods for processing experimental data in physical sciences.",
+    "quant-ph": "Quantum Physics: Explores quantum computing, quantum information theory, and quantum system implementations.",
+    "cs.IR": "Information Retrieval: Designs systems for effective search and organization of large-scale document collections.",
+    "math.NA": "Numerical Mathematics: Analyzes mathematical foundations of numerical algorithms for differential equations and linear algebra.",
+    "math.PR": "Probability Theory: Studies random phenomena, stochastic processes, and uncertainty quantification.",
+    "stat.ME": "Statistical Methods: Develops new techniques for experimental design, data collection, and analysis methodologies.",
+    "cs.SE": "Software Engineering: Addresses systematic approaches to software development, maintenance, and quality assurance.",
+    "math.OC": "Optimization Theory: Formulates and solves mathematical optimization problems with applications in logistics and operations research.",
+    "math.IT": "Mathematical Information Theory: Provides rigorous foundations for data transmission, compression, and coding theory.",
+    "cs.HC": "Human-Computer Interaction: Designs intuitive interfaces and studies user behavior with computing systems.",
+    "stat.TH": "Theoretical Statistics: Establishes mathematical foundations for statistical inference and decision-making processes.",
+    "cs.NI": "Network Systems: Architectures communication protocols and internet infrastructure for reliable data transmission.",
+    "cs.CV": "Computer Vision: Enables machines to interpret and understand visual information from images and video.",
+    "cs.SD": "Sound Processing: Analyzes and synthesizes audio signals for applications in speech recognition and music information retrieval."
+}