patcg-individual-drafts / topics

The Topics API
https://patcg-individual-drafts.github.io/topics/
Other
589 stars 168 forks source link

Invalid "/" characters for 625 domains in the override list v4 of the Topics API #306

Closed yohhaan closed 2 months ago

yohhaan commented 3 months ago

Hello,

I am re-opening here a bug I had initially reported through the Chrome Platform Status as it remained unanswered there but the issue appears to still be present as far as I can tell.

Problem Description

Google ships with Chrome an override list, which is a mapping manually annotated by Google of domains to their corresponding topics. When a domain is being classified by the Topics API, it is checked against that list first and if not found classified by the ML model.

The override list shipped in Chrome with the 4th version of the Topics model contains 47 128 such individual mappings from domains to topics. However, it appears that 625 of them are not formatted correctly as they contain the "/" invalid character (according to the pre-processing rules applied to the input) and are flipped around that invalid character. As a result, when trying to classify the corresponding initial domain by applying the pre-processing to the input domain, no match is found in the override list. Thus, the ML model is used for the classification and different results than the manual annotations are output.

Overview and some more details on this blog post

Steps to reproduce

Here are some steps you can follow to reproduce the issue and the misclassifications due to the incorrectly formatted entries in the override list of the Topics API feature.

  1. Get the override_list.pb.gz shipped with model v4 (current version of the model.tflite for the Topics API shipped in Chrome).

    • Visit chrome://topics-internals, under the Classifier tab, get the path where the model is stored, in that same folder you will find the override_list.pb.gz file
    • For archival purposes, the file can also be found here
  2. Then, check that the override list contains domains with the invalid "/" character and that these domains are flipped around that character, you will need to decompress the protobuf file, get the corresponding .proto, and decode it. Here is how I do it, you will need these 2 scripts:

    convert_pb_override.sh bash script:

    #!/bin/bash
    
    override_pb_gz=$1
    override_tsv=$2
    
    override_pb=override.pb
    proto_path=page_topics_override_list.proto
    python_proto_path=page_topics_override_list_pb2.py
    if [ ! -f $override_tsv ]
    then
        # Fetch page_topics_override_list.proto
        wget -q -O $proto_path https://raw.githubusercontent.com/chromium/chromium/main/components/optimization_guide/proto/page_topics_override_list.proto
        protoc $proto_path --python_out=.
        # Decompress override.pb.gz
        gzip -cdk $override_pb_gz > $override_pb
        python3 convert_pb_override.py $override_pb > $override_tsv
        rm $proto_path $override_pb $python_proto_path
    fi

    convert_pb_override.py python script:

    import argparse
    import pandas as pd
    import page_topics_override_list_pb2
    
    # Create Argument Parser
    parser = argparse.ArgumentParser(
        prog="python3 convert_pb_override.py",
        description="Convert .pb override list to .tsv",
    )
    parser.add_argument("input_file", help="input file")
    args = parser.parse_args()
    
    # Load override list
    override_list = page_topics_override_list_pb2.PageTopicsOverrideList()
    with open(args.input_file, "rb") as f:
        override_list.ParseFromString(f.read())
    print("domain\ttopics")
    for entry in override_list.entries:
        line = "{}".format(entry.domain)
        first_topic = True
        for id in entry.topics.topic_ids:
            if first_topic:
                line += "\t{}".format(id)
                first_topic = False
            else:
                line += ",{}".format(id)
        print(line)
    • And then run:
      # Decode override list to .tsv format
      ./convert_pv_override.sh override_list.pb.gz override_list.tsv
      # Extract domains (and corresponding topics) with invalid character:
      grep ".*[^[:alpha:][:space:][:digit:]^,].*" override_list.tsv

The domain entries in that override list are supposed to be pre-processed the same way as the input that would be passed to the model.tflite of the Topics API. This means: take the FQDN, remove any "www." prefix if present in the domain to classify, and then replace the following characters "-", "_", ".", "+" by a whitespace (https://source.chromium.org/chromium/chromium/src/+/main:components/browsing_topics/annotator_impl.cc;l=269).

Some examples:

As a result, when these domains are classified by the Topics API in Chrome, no match is found in the override list for the domain correctly pre-processed. Thus, they are classified by the ML model which does not output the intended classification (Chrome classification can be obtained from chrome://topics-internals):

yohhaan commented 2 months ago

Closing as it has been fixed, see the initial issue for more details: https://issues.chromium.org/issues/325123734