search
HomeTechnology peripheralsAIProfessor Ma Yi's new work: White-box ViT successfully achieves 'partitioned emergence', is the era of empirical deep learning coming to an end?

The basic visual model based on Transformer has shown very powerful performance in various downstream tasks, such as segmentation and detection, and models such as DINO have emerged with semantic segmentation attributes after self-supervised training.

It is strange that the visual Transformer model does not have similar emergent capabilities after being trained for supervised classification

Recently, Professor Ma Yi's team studied models based on the Transformer architecture to explore whether the emergent segmentation ability is simply the result of a complex self-supervised learning mechanism, or whether the same performance can be achieved under more general conditions by appropriately designing the model architecture. Emergence

Professor Ma Yis new work: White-box ViT successfully achieves partitioned emergence, is the era of empirical deep learning coming to an end?

Code link: https://github.com/Ma-Lab-Berkeley/CRATE

Please click the following link to view the paper: https://arxiv.org/abs/2308.16271

After a large number of experiments, the researchers proved that when using the white-box Transformer model CRATE, its design Explicitly model and pursue low-dimensional structure in data distributions to emerge whole- and part-level segmentation properties with minimally supervised training recipes

Through hierarchical fine-grained analysis, we get An important conclusion is drawn: emergent properties strongly confirm the design mathematical function of white-box networks. Based on this result, we proposed a method to design a white-box basic model that is not only high-performance but also fully mathematically interpretable

Professor Ma Yi also said, Research on deep learning will gradually shift from empirical design to theoretical guidance.

Professor Ma Yis new work: White-box ViT successfully achieves partitioned emergence, is the era of empirical deep learning coming to an end?

The emergent properties of white-box CRATE

DINO’s segmentation-emergent ability refers to the DINO model’s ability to process language tasks At this time, the input sentence can be divided into smaller fragments and each fragment can be processed independently. This ability enables the DINO model to better understand complex sentence structures and semantic information, thereby improving its performance in the field of natural language processing

Representation learning in intelligent systems It aims to convert the world's high-dimensional, multi-modal sensory data (images, language, speech) into a more compact form while retaining its basic low-dimensional structure to achieve efficient recognition (such as classification) and grouping (such as segmentation) and tracking.

Professor Ma Yis new work: White-box ViT successfully achieves partitioned emergence, is the era of empirical deep learning coming to an end?

The training of deep learning models usually adopts a data-driven approach, by inputting large-scale data and learning in a self-supervised manner

Among the basic vision models, the DINO model shows surprising emergence capabilities. ViTs can recognize explicit semantic segmentation information even without supervised segmentation training. The DINO model with self-supervised Transformer architecture performs well in this regard

Follow-up work has studied how to utilize this segmentation information in the DINO model and perform it in downstream tasks such as segmentation and detection. etc. have achieved state-of-the-art performance, and some work has also proven that the penultimate layer features in ViTs trained with DINO are strongly related to the salience information in the visual input, such as distinguishing foreground, background and object boundaries, thereby improving image segmentation and other task performance.

In order to highlight the segmentation attributes, DINO needs to skillfully combine self-supervised learning, knowledge distillation and weight averaging methods during the training process

It is unclear whether each component introduced in DINO is essential for the emergence of segmentation masks. Although DINO also adopts the ViT architecture as its backbone, in the ordinary supervised ViT model trained on the classification task, and No segmentation emergent behavior was observed.

The emergence of CRATE

Based on the successful case of DINO, researchers want to explore complex self-supervised learning Is the pipeline necessary to obtain emergent properties in Transformer-like visual models?

The researchers believe that a promising way to promote segmentation properties in Transformer models is to design the Transformer model architecture taking into account the input data structure, which also represents representation learning. A combination of classic methods and modern data-driven deep learning frameworks.

Professor Ma Yis new work: White-box ViT successfully achieves partitioned emergence, is the era of empirical deep learning coming to an end?

Compared with the current mainstream Transformer model, this design method can also be called a white-box Transformer model.

Based on the previous work of Professor Ma Yi’s group, researchers conducted extensive experiments on the CRATE model with a white-box architecture, proving that CRATE’s white-box design is the reason for the emergence of segmentation attributes in self-attention graphs.

What needs to be rephrased is: Qualitative assessment

The researcher uses the [CLS] token-based attention Strive to explain and visualize the model using the graph method and find that the query-key-value matrices in CRATE are all the same

Professor Ma Yis new work: White-box ViT successfully achieves partitioned emergence, is the era of empirical deep learning coming to an end?

The CRATE model can be observed The self-attention map can correspond to the semantics of the input image. The internal network of the model performs clear semantic segmentation on each image, achieving an effect similar to the DINO model.

Ordinary ViT does not show similar segmentation properties when trained on supervised classification tasks

Professor Ma Yis new work: White-box ViT successfully achieves partitioned emergence, is the era of empirical deep learning coming to an end?

Based on previous research on visual image learning of block-by-block deep features, researchers conducted principal component analysis (PCA) research on deep token representations of CRATE and ViT models

Professor Ma Yis new work: White-box ViT successfully achieves partitioned emergence, is the era of empirical deep learning coming to an end?

It can be found that CRATE can still capture the boundaries of objects in the image without segmentation supervision training.

Moreover, the principal components also indicate the feature alignment of similar parts between the token and the object, such as the red channel corresponding to the horse’s leg

The degree of PCA visualization structure of the supervised ViT model is quite low.

Quantitative Evaluation

The researchers used existing segmentations to evaluate the CRATE emergent segmentation properties. and object detection technology

As can be seen from the self-attention map, CRATE explicitly captures object-level semantics with clear boundaries. In order to quantitatively measure the quality of segmentation, the researchers used self-attention to The attention map generates a segmentation mask and compares the standard mIoU (mean intersection-over-union ratio) with the real mask.

Professor Ma Yis new work: White-box ViT successfully achieves partitioned emergence, is the era of empirical deep learning coming to an end?

It can be seen from the experimental results that CRATE is significantly better than ViT in terms of visual and mIOU scores, which shows that the internal representation of CRATE is useful for generating segmentation masks. Mask tasks more efficiently

Object detection and fine-grained segmentation

To further validate and evaluate the rich semantics captured by CRATE Information, the researchers adopted MaskCut, an efficient object detection and segmentation method, to obtain an automated evaluation model without manual annotation, and can extract more fine-grained segmentation from images based on the token representation learned by CRATE.

Professor Ma Yis new work: White-box ViT successfully achieves partitioned emergence, is the era of empirical deep learning coming to an end?

Professor Ma Yis new work: White-box ViT successfully achieves partitioned emergence, is the era of empirical deep learning coming to an end?

As you can see in the segmentation results on COCO val2017, there is an internal representation of CRATE in the detection and The segmentation index is better than that of supervised ViT. MaskCut with supervised ViT features cannot even produce segmentation masks at all in some cases.

White box analysis of CRATE’s segmentation capabilities

The role of depth in CRATE

The design of each layer of CRATE follows the same conceptual purpose: to optimize sparse rate reduction and transform token distribution into a compact and structured form. After rewriting: The design of each level of CRATE follows the same philosophy: optimize the reduction of sparse rate, and transform the distribution of tokens into a compact and structured form

Assume CRATE The emergence of semantic segmentation capabilities is similar to "representing clusters of tokens belonging to similar semantic categories in Z". It is expected that CRATE's segmentation performance can improve with increasing depth.

To test this, the researchers used the MaskCut pipeline to quantitatively evaluate segmentation performance across internal representations across different layers; and also applied PCA visualization to understand how segmentation emerges with depth. .

Professor Ma Yis new work: White-box ViT successfully achieves partitioned emergence, is the era of empirical deep learning coming to an end?

It can be observed from the experimental results that the segmentation score improves when using representations from deeper layers, which is very consistent with CRATE’s incremental optimization design.

In contrast, even though the performance of ViT-B/8 improves slightly in later layers, its segmentation scores are significantly lower than CRATE. The PCA results show that deep extraction from CRATE The representation will gradually pay more attention to foreground objects and be able to capture texture-level details.

melting experiment in CRATE

Attention block (MSSA) in CRATE and The MLP block (ISTA) is different from the attention block in ViT

To study the impact of each component on the emergent segmentation properties of CRATE, the researchers selected three CRATE variants: CRATE, CRATE-MHSA, CRATE-MLP. These variants represent the attention block (MHSA) and the MLP block in ViT respectively

The researchers applied the same pre-training settings on the ImageNet-21k dataset and then applied coarse segmentation evaluation and mask segmentation evaluation to quantitatively compare the performance of different models.

Professor Ma Yis new work: White-box ViT successfully achieves partitioned emergence, is the era of empirical deep learning coming to an end?

According to experimental results, CRATE significantly outperforms other model architectures in all tasks. It is worth noting that although the architectural difference between MHSA and MSSA is small, simply replacing MHSA in ViT with MSSA in CRATE can significantly improve ViT's coarse segmentation performance (i.e. VOC Seg). Performance. This further proves the effectiveness of white-box design

The content that needs to be rewritten is: pay attention to the identification of the semantic attributes of the header

[CLS] The self-attention map between the token and the image block token can see a clear segmentation mask. According to intuition, each attention head should be able to capture some features of the data.

The researchers first input the image into the CRATE model, and then had a human inspect and select four attention heads that seemed to have semantic meaning; then they used these attention heads on other input images. Self-attention map visualization on the head.

Professor Ma Yis new work: White-box ViT successfully achieves partitioned emergence, is the era of empirical deep learning coming to an end?

Observation can be found that each attention head can capture different parts of the object, and even different semantics. For example, the attention head in the first column can capture the legs of different animals, while the attention head in the last column can capture the ears and head

Since it is deformable This ability to parse visual input into part-whole hierarchies has been a goal of recognition architecture since the release of deformable part models and capsule networks, and the CRATE model of white-box design also has this ability.

The above is the detailed content of Professor Ma Yi's new work: White-box ViT successfully achieves 'partitioned emergence', is the era of empirical deep learning coming to an end?. For more information, please follow other related articles on the PHP Chinese website!

Statement
This article is reproduced at:51CTO.COM. If there is any infringement, please contact admin@php.cn delete
AI Therapists Are Here: 14 Groundbreaking Mental Health Tools You Need To KnowAI Therapists Are Here: 14 Groundbreaking Mental Health Tools You Need To KnowApr 30, 2025 am 11:17 AM

While it can’t provide the human connection and intuition of a trained therapist, research has shown that many people are comfortable sharing their worries and concerns with relatively faceless and anonymous AI bots. Whether this is always a good i

Calling AI To The Grocery AisleCalling AI To The Grocery AisleApr 30, 2025 am 11:16 AM

Artificial intelligence (AI), a technology decades in the making, is revolutionizing the food retail industry. From large-scale efficiency gains and cost reductions to streamlined processes across various business functions, AI's impact is undeniabl

Getting Pep Talks From Generative AI To Lift Your SpiritGetting Pep Talks From Generative AI To Lift Your SpiritApr 30, 2025 am 11:15 AM

Let’s talk about it. This analysis of an innovative AI breakthrough is part of my ongoing Forbes column coverage on the latest in AI including identifying and explaining various impactful AI complexities (see the link here). In addition, for my comp

Why AI-Powered Hyper-Personalization Is A Must For All BusinessesWhy AI-Powered Hyper-Personalization Is A Must For All BusinessesApr 30, 2025 am 11:14 AM

Maintaining a professional image requires occasional wardrobe updates. While online shopping is convenient, it lacks the certainty of in-person try-ons. My solution? AI-powered personalization. I envision an AI assistant curating clothing selecti

Forget Duolingo: Google Translate's New AI Feature Teaches LanguagesForget Duolingo: Google Translate's New AI Feature Teaches LanguagesApr 30, 2025 am 11:13 AM

Google Translate adds language learning function According to Android Authority, app expert AssembleDebug has found that the latest version of the Google Translate app contains a new "practice" mode of testing code designed to help users improve their language skills through personalized activities. This feature is currently invisible to users, but AssembleDebug is able to partially activate it and view some of its new user interface elements. When activated, the feature adds a new Graduation Cap icon at the bottom of the screen marked with a "Beta" badge indicating that the "Practice" feature will be released initially in experimental form. The related pop-up prompt shows "Practice the activities tailored for you!", which means Google will generate customized

They're Making TCP/IP For AI, And It's Called NANDAThey're Making TCP/IP For AI, And It's Called NANDAApr 30, 2025 am 11:12 AM

MIT researchers are developing NANDA, a groundbreaking web protocol designed for AI agents. Short for Networked Agents and Decentralized AI, NANDA builds upon Anthropic's Model Context Protocol (MCP) by adding internet capabilities, enabling AI agen

The Prompt: Deepfake Detection Is A Booming BusinessThe Prompt: Deepfake Detection Is A Booming BusinessApr 30, 2025 am 11:11 AM

Meta's Latest Venture: An AI App to Rival ChatGPT Meta, the parent company of Facebook, Instagram, WhatsApp, and Threads, is launching a new AI-powered application. This standalone app, Meta AI, aims to compete directly with OpenAI's ChatGPT. Lever

The Next Two Years In AI Cybersecurity For Business LeadersThe Next Two Years In AI Cybersecurity For Business LeadersApr 30, 2025 am 11:10 AM

Navigating the Rising Tide of AI Cyber Attacks Recently, Jason Clinton, CISO for Anthropic, underscored the emerging risks tied to non-human identities—as machine-to-machine communication proliferates, safeguarding these "identities" become

See all articles

Hot AI Tools

Undresser.AI Undress

Undresser.AI Undress

AI-powered app for creating realistic nude photos

AI Clothes Remover

AI Clothes Remover

Online AI tool for removing clothes from photos.

Undress AI Tool

Undress AI Tool

Undress images for free

Clothoff.io

Clothoff.io

AI clothes remover

Video Face Swap

Video Face Swap

Swap faces in any video effortlessly with our completely free AI face swap tool!

Hot Tools

ZendStudio 13.5.1 Mac

ZendStudio 13.5.1 Mac

Powerful PHP integrated development environment

MantisBT

MantisBT

Mantis is an easy-to-deploy web-based defect tracking tool designed to aid in product defect tracking. It requires PHP, MySQL and a web server. Check out our demo and hosting services.

SecLists

SecLists

SecLists is the ultimate security tester's companion. It is a collection of various types of lists that are frequently used during security assessments, all in one place. SecLists helps make security testing more efficient and productive by conveniently providing all the lists a security tester might need. List types include usernames, passwords, URLs, fuzzing payloads, sensitive data patterns, web shells, and more. The tester can simply pull this repository onto a new test machine and he will have access to every type of list he needs.

Notepad++7.3.1

Notepad++7.3.1

Easy-to-use and free code editor

DVWA

DVWA

Damn Vulnerable Web App (DVWA) is a PHP/MySQL web application that is very vulnerable. Its main goals are to be an aid for security professionals to test their skills and tools in a legal environment, to help web developers better understand the process of securing web applications, and to help teachers/students teach/learn in a classroom environment Web application security. The goal of DVWA is to practice some of the most common web vulnerabilities through a simple and straightforward interface, with varying degrees of difficulty. Please note that this software