APC 技術ブログ


株式会社 エーピーコミュニケーションズの技術ブログです。

A Practical Introduction to Machine Learning with Databricks Mosaic AI


Welcome to "A Practical Introduction to Machine Learning Using Databricks Mosaic AI." Today, we are thrilled to host Craig, who oversees AI and machine learning product management at Databricks, to share insights into this burgeoning field. This session offers a perfect opportunity to revisit key machine learning themes that have been prevalent before November 2022, and promises to be equally exhilarating for both speakers and the audience.

Machine learning involves the process of learning from data and leveraging this knowledge to make informed predictions about new data. Its applications are vast, ranging from enhancing product recommendations in e-commerce to assisting medical diagnoses in the healthcare industry. Technological advances like Databricks Mosaic AI have dramatically evolved real-time inference capabilities. Today, businesses and researchers can process extensive datasets almost instantaneously, enabling faster data-driven decisions.

The technological advancements made through Databricks Mosaic AI not only improve business operations but also significantly impact our daily lives. The importance of machine learning and real-time data processing continues to grow, affecting various sectors and demonstrating the profound influence of AI in modern life.

The goal of this session is to deepen the understanding of machine learning fundamentals and explore the transformative potential of industries through innovative AI applications like Databricks Mosaic AI.

Governance, Provenance, and Integrated Data Stack

Data governance and provenance are essential for the effective execution of massive AI initiatives. Through meticulous management, data is handled with consistency and transparency, aiding businesses in regulatory compliance and risk management. Databricks Mosaic AI plays a crucial role in this framework by providing an integrated track from data ingestion to business insights, thus consolidating operations.

In the realm of data governance, systems are designed to define and control who accesses the data and detail how and when it is used. Such structured oversight enhances data security and ensures adherence to compliance standards. Meanwhile, data provenance offers a detailed record of data evolution, maintaining information integrity and reliability.

Furthermore, the integrated data stack provided by Databricks Mosaic AI centralizes data management across different sources. This consolidation simplifies complex data transactions and integrations across diverse systems, accelerating the decision-making process for data scientists and engineers.

This robust infrastructure not only facilitates seamless data utilization but also fosters innovation and prepares the foundation for advanced data strategies and applications in an AI-driven landscape. These efforts align closely with the objectives of Databricks Mosaic AI, significantly impacting the operational efficiency and strategic capabilities of enterprises engaged in artificial intelligence and data science.

Model Training and Management Using Databricks Mosaic AI

The workshop on Databricks Mosaic AI highlighted the initial stages of the model training process, emphasizing the importance of data preprocessing using Data Lake Transformation (DLT). This step is crucial for preparing the data and ultimately storing it in feature tables, setting the stage for model training.

Databricks offers two main methodologies for training models. The mainstream approach is the "code-first" method, where users engage in writing code using Databricks' notebooks. This environment supports a multitude of libraries, including TensorFlow, PyTorch, XGBoost, and Scikit-learn, providing flexibility to meet diverse project requirements.

The entire model-building process is meticulously documented within these notebooks. Every action and modification is recorded, providing detailed blueprints of what was done and how changes might be reversed. This level of detailed tracking is invaluable for fine-tuning models and ensuring reproducibility across experiments.

This platform not only simplifies the process of model training and management but also enhances the efficiency and effectiveness of these processes. The session highlighted how Databricks' robust and adaptable tools are crucial in streamlining the transition from data preprocessing to model training, facilitating more practical and impactful machine learning implementations.

The session focused on deploying machine learning using Databricks Mosaic AI emphasized real-time inference capabilities and advanced feature delivery technologies, highlighting methods for efficiently and instantly using AI models and data. Notably, segments related to MLOps and continuous improvement processes were emphasized.

MLOps and Continuous Improvement Process

The MLOps process begins with model performance management, incorporating steps related to regular model retraining and deployment. This process adheres to the following steps:

  1. Integration of Feature Engineering and ML Training Pipelines: Initial feature extraction and integration into the training pipeline enable more accurate model training.

  2. Testing and Validation Checks: After a model is trained, verifying its accuracy with an appropriate test set is crucial. This stage is vital to ensure the model performs well in real scenarios.

  3. Regular Retraining and Model Validation: To reflect new data and parameters, models are retrained regularly, continuously enhancing their effectiveness.

  4. Incremental Model Deployment: Instead of handling all traffic immediately, models begin deployment with a small percentage of traffic (e.g., 10% or 20%) and observe performance.

  5. Operational Testing in Shadow Environments: Before handling all actual traffic, models are tested in shadow environments to evaluate performance in production settings.

Adhering to these steps promotes continuous improvement and proper management of models, enabling organizations to effectively utilize AI models. MLOps is an essential practice for managing successful machine learning projects.

Real-Time Inference and the Future Outlook of Databricks Mosaic AI

This session provided an opportunity to explore Databricks' cutting-edge technologies and focused on the utilization of Databricks Mosaic AI, mainly discussing real-time inference capabilities and potential future enhancements.

Importance of Real-Time Inference

The concept of "inference tables" plays a crucial role in real-time inference. Utilizing these inference tables, each query log to the model is stored, documenting what was queried and the outcomes obtained. This data is immensely helpful in understanding model evolution over time.

  • Feature Drift: Understands how features input through the model change over time.
  • Model Prediction Changes: Tracks how predictions evolve.

Based on this information, it becomes possible to identify when models need to be retrained.

Model Implementation and Applications

The session also touched on how models are used after deployment. Models can be queried directly from a browser or integrated into third-party frontends or Python environments via APIs. This facilitates easy application across various applications and serves as a foundation for generating new features.

The real-time inference technologies of Databricks Mosaic AI hold significant potential to dramatically alter the future of AI model and data usage. This session allowed a deeper understanding of its sophistication and broad application scope through concrete examples. Keeping an eye on the evolution of this technology is crucial for the industry, as its advancements could ignite new innovations.


As explored in today's session "Introduction to Machine Learning with Databricks Mosaic AI," real-time inference is not just a tool but a transformative force in artificial intelligence. Updates and enhancements to Databricks Mosaic AI will further expand the boundaries of what can be achieved with real-time data processing. As the technology matures, embracing these developments will become crucial for companies and professionals striving to remain competitive in a data-driven world. Hence, continuous learning and adaptation alongside growing tech trends are essential.

About the special site during DAIS

This year, we have prepared a special site to report on the session contents and the situation from the DAIS site! We plan to update the blog every day during DAIS, so please take a look.