Full Unicode Search at 50× ICU Speed with AVX‑512

(ashvardanian.com)

95 points | by ashvardanian 23 hours ago

11 comments

  • bbminner 16 minutes ago
    I was really confused about the case folding, this page explained the motivation well https://jean.abou-samra.fr/blog/unicode-misconceptions

    """ Continuing with the previous example of “ß”, one has lowercase("ss") != lowercase("ß") but uppercase("ss") == uppercase("ß"). Conversely, for legacy reasons (compatibility with encodings predating Unicode), there exists a Kelvin sign “K”, which is distinct from the Latin uppercase letter “K”, but also lowercases to the normal Latin lowercase letter “k”, so that uppercase("K") != uppercase("K") but lowercase("K") == lowercase("K").

    The correct way is to use Unicode case folding, a form of normalization designed specifically for case-insensitive comparisons. Both casefold("ß") == casefold("ss") and casefold("K") == casefold("K") are true. Case folding usually yields the same result as lowercasing, but not always (e.g., “ß” lowercases to itself but case-folds to “ss”). """

    One question I have is why have Kelvin sign that is distinct from Latin K and other indistinguishable symbols? To make quantified machine readable (oh, this is not a 100K license plate or money amount, but a temperature)? Or to make it easier for specialized software to display it in correct placed/units?

    • bee_rider 5 minutes ago
      They seem to have (if I understand correctly) degree-Celsius and degree-Fahrenheit symbols. So maybe Kelvin is included for consistency, and it just happens to look identical to Latin K?

      IMO the confusing bit is giving it a lower case. It is a symbol that happens to look like an upper case, not an actual letter…

  • anonnon 3 minutes ago
    > ICU has bindings for Rust that provide case-folding functionality, but not case-insensitive substring search.

    > ICU has many bindings. The Rust one doesn’t expose any substring search functionality, but the Python one does:

    The Python's ICU support is based on ICU4C. Rust's ICU "bindings" are actually a new implementation called ICU4X.

  • unwind 2 hours ago
    Very cool and impressive performance.

    I was worried (I find it confusing when Unicode "shadows" of normal letters exist, and those are of course also dangerous in some cases when they can be mis-interpreted for the letter they look more or less exactly like) by the article's use of U+212A (Kelvin symbol) as sample text, so I had to look it up [1].

    Anyway, according to Wikipedia the dedicated symbol should not be used:

    However, this is a compatibility character provided for compatibility with legacy encodings. The Unicode standard recommends using U+004B K LATIN CAPITAL LETTER K instead; that is, a normal capital K.

    That was comforting, to me. :)

    [1]: https://en.wikipedia.org/wiki/Kelvin#Orthography

    • jjmarr 2 hours ago
      > I find it confusing when Unicode "shadows" of normal letters exist, and those are of course also dangerous in some cases when they can be mis-interpreted for the letter they look more or less exactly like

      Isn't this why Unicode normalization exists? This would let you compare Unicode letters and determine if they are canonically equivalent.

      • Sesse__ 43 minutes ago
        It's why the Unicode Collation Algorithm exists.

        If you look in allkeys.txt (the base UCA data, used if you don't have language-specific stuff in your comparisons) for the two code points in question, you'll find:

          004B  ; [.2514.0020.0008] # LATIN CAPITAL LETTER K
          212A  ; [.2514.0020.0008] # KELVIN SIGN
        
        The numbers in the brackets are values on level 1 (base), level 2 (typically used for accents), level 3 (typically used for case). So they are to compare identical under the UCA, in almost every case except for if you really need a tiebreaker.

        Compare e.g. :

          1D424 ; [.2514.0020.0005] # MATHEMATICAL BOLD SMALL K
        
        which would compare equal to those under a case-insensitive accent-sensitive collation, but _not_a case-sensitive one (case-sensitive collations are always accent-sensitive, too).
      • ComputerGuru 1 hour ago
        Normalization wouldn’t address this.
        • happytoexplain 1 hour ago
          What do you mean? All four normal forms of the Kelvin 'K' are the Latin 'K', as far as I can tell.
        • nwellnhof 1 hour ago
          Normalization forms NFKC and NFKD that also handle compatibility equivalence do.
          • mananaysiempre 48 minutes ago
            A few deprecated characters, including the Kelvin and Ångström symbols, are in fact canonically equivalent to their replacements and not just compatibility equivalent, so plain NFC/NFD is enough. (It’s generally better to avoid NFKC/NFKD normalizations unless you fully understand the implications, as they do lose meaning and at the same time do not account for all possible confusables.)
  • hans_castorp 33 minutes ago
    Would be cool if that could be integrated into Postgres :)
    • ashvardanian 17 minutes ago
      I was just about to ask some friends about it. If I’m not mistaken, Postgres began using ICU for collation, but not string matching yet. Curious if someone here is working in that direction?
  • orthoxerox 2 hours ago
    Is it possible to extend this to support additional transformation rules like Any-Latin;Latin-ASCII? To make it possible to find "Վարդանյան" in a haystack by searching for "vardanyan"?
    • ashvardanian 2 hours ago
      Yes — fuzzy and phonetic matching across languages is part of the roadmap. That space is still poorly standardized, so I wanted to start with something widely understood and well-defined (ICU-style transforms) before layering on more advanced behavior.

      Also, as shown in the later tables, the Armenian and Georgian fast paths still have room for improvement. Before introducing higher-level APIs, I need to tighten the existing Armenian kernel and add a dedicated one for Georgian. It’s not a true bicameral script, but some characters are folding fold targets for older scripts, which currently forces too many fallbacks to the serial path.

  • ashvardanian 2 hours ago
    This article is about the ugliest — but arguably the most important — piece of open-source software I’ve written this year. The write-up ended up long and dense, so here’s a short TL;DR:

    I grouped all Unicode 17 case-folding rules and built ~3K lines of AVX-512 kernels around them to enable fully standards-compliant, case-insensitive substring search across the entire 1M+ Unicode range, operating directly on UTF-8 bytes. In practice, this is often ~50× faster than ICU, and also less wrong than most tools people rely on today—from grep-style utilities to products like Google Docs, Microsoft Excel, and VS Code.

    StringZilla v4.5 is available for C99, C++11, Python 3, Rust, Swift, Go, and JavaScript. The article covers the algorithmic tradeoffs, benchmarks across 20+ Wikipedia dumps in different languages, and quick starts for each binding.

    Thanks to everyone for feature requests and bug reports. I'll do my best to port this to Arm as well — but first, I'm trying to ship one more thing before year's end.

    • adzm 1 hour ago
      This is a truly amazing accomplishment. Reading these kernels is a joy!
    • fatty_patty89 2 hours ago
      Thank you

      do the go bindings require cgo?

      • ashvardanian 2 hours ago
        The GoLang bindings – yes, they are based on cGo. I realize it's suboptimal, but seems like the only practical option at this point.
        • fatty_patty89 2 hours ago
          In a normal world the Go C FFI wouldn't have insane overhead but what can we do, the language is perfect and it will stay that way until morale improves.

          Thanks for the work you do

          • kbolino 23 minutes ago
            There are undoubtedly still some optimizations lying around, but the biggest source of Go's FFI overhead is goroutines.

            There's only two "easy" solutions I can see: switch to N:N threading model or make the C code goroutine-aware. The former would speed up C calls at the expense of slowing down lots of ordinary Go code. Personally, I can still see some scenarios where that's beneficial, but it's pretty niche. The latter would greatly complicate the use of cgo, and defeat one of its core purposes, namely having access to large hard-to-translate C codebases without requiring extensive modifications of them.

            A lot of people compare Go's FFI overhead to that of other natively compiled languages, like Zig or Rust, or to managed runtime languages like Java (JVM) or C# (.NET), but none of those alternatives use green threads (the general concept behind goroutines) at all or as extensively. If you really want to compare apples-to-apples, you should compare against Erlang (BEAM). As far as I can tell, Erlang NIFs [1] are broadly similar to purego [2] calls, and their runtime performance [3] has more or less the same issues as CGo [4].

            [1]: https://www.erlang.org/doc/system/nif.html

            [2]: https://pkg.go.dev/github.com/ebitengine/purego

            [3]: https://erlang.org/documentation/doc-10.1/doc/efficiency_gui...

            [4]: https://www.reddit.com/r/golang/comments/12nt2le/when_dealin...

          • kardianos 1 hour ago
            In a real (not "normal") world, trade-offs exist and Go choose a specific set of design points that are consequential.
  • moralestapia 40 minutes ago
    Ash is amazing!

    Also very cool and approachable guy.

    (Best wishes if you're reading this.)

  • mgaunard 3 hours ago
    In practice you should always normalize your Unicode data, then all you need to do is memcmp + boundary check.

    Interestingly enough this library doesn't provide grapheme cluster tokenization and/or boundary checking which is one of the most useful primitive for this.

    • stingraycharles 3 hours ago
      That’s not practical in many situations, as the normalization alone may very well be more expensive than the search.

      If you’re in control of all data representations in your entire stack, then yes of course, but that’s hardly ever the case and different tradeoffs are made at different times (eg storage in UTF-8 because of efficiency, but in-memory representation in UTF-32 because of speed).

      • mgaunard 3 hours ago
        That doesn't make sense; the search is doing on-the-fly normalization as part of its algorithm, so it cannot be faster than normalization alone.
        • ashvardanian 2 hours ago
          I get why it sounds that way, but it’s not actually true.

          StringZilla added full Unicode case folding in an earlier release, and had a state-of-the-art exact case-sensitive substring search for years. However, doing a full fold of the entire haystack is significantly slower than the new case-insensitive search path.

          The key point is that you don’t need to fully normalize the haystack to correctly answer most substring queries. The search algorithm can rule out the vast majority of positions using cheap, SIMD-friendly probes and only apply fold logic on a very small subset of candidates.

          I go into the details in the “Ideation & Challenges in Substring Search” section of the article

        • Const-me 2 hours ago
          > it cannot be faster than normalization alone

          Modern processors are generally computing stuff way faster than they can load and store bytes from main memory.

          The code which does on the fly normalization only needs to normalize a small window. If you’re careful, you can even keep that window in registers, which have single CPU cycle access latency and ridiculously high throughput like 500GB/sec. Even if you have to store and reload, on-the-fly normalization is likely to handle tiny windows which fit in the in-core L1D cache. The access cost for L1D is like ~5 cycles of latency, and equally high throughput because many modern processors can load two 64-bytes vectors and store one vector each and every cycle.

          • mgaunard 1 hour ago
            The author published the bandwidth of its algo, it's one fifth of a typical memory bandwidth (it's not possible to go faster than memory obviously for this benchmark, since we're assuming the data is not in cache).
        • stingraycharles 2 hours ago
          It can, because of how CPUs work with registers and hot code paths and all that.

          First normalizing everything and then comparing normalized versions isn’t as fast.

          And it also enables “stopping early” when a match has been found / not found, you may not actually have to convert everything.

          • mgaunard 1 hour ago
            Running more code per unit of data does not make the code hotter or reduce the register pressure, quite the opposite...
            • stingraycharles 1 hour ago
              You’re misunderstanding: you just convert to 32 bits once and reuse that same register all the time.

              You’re running the exact same code, but are more more efficient in terms of “I immediately use the data for comparison after converting it”, which means it’s likely either in a register or L1 cache already.

    • orthoxerox 3 hours ago
      In practice the data is not always yours to normalize. You're not going to case-fold your library, but you still want to be able to search it.
  • kardianos 1 hour ago
    Looks neat. What are all the genomic sequence comparisons in there for? Is this a grab bag of interesting string methods or is there a motivation for this?
    • ashvardanian 1 hour ago
      Levenshtein distance calculations are a pretty generic string operation, Genomics happens to be one of the domains where they are most used... and a passion of mine :)
  • xking5205 2 hours ago
    its good
  • andersa 3 hours ago
    From a German user perspective, ICU and your fancy library are incorrect, actually. Mass is not a different casing of Maß, they are different characters. Google likely changed this because it didn't do what users wanted.
    • pjmlp 2 hours ago
      It isn't until it is, how would you write it when ß isn't available on the keyboard?

      Which is why we also have to deal with the ue, ae, oe kind of trick, also known as Ersatzschreibweise.

      Then German language users from de-CH region, consider Mass the correct way.

      Yeah, localization and internalization is a mess to get right.

      • wat10000 2 hours ago
        Case insensitivity is localized like anything else. I and i are equivalent, right? Not if you’re doing Turkish, then it’s I and ı, and İ and i.

        In practice you can do pretty well with a universal approach, but it can’t be 100% correct.

        • ashvardanian 1 hour ago
          This is a very good example! Still, “correct” needs context. You can be 100% “correct with respect to ICU”. It’s definitely not perfect, but it’s the best standard we have. And luckily for me, it also defines the locale-independent rules. I can expand to support locale-specific adjustments in the future, but waiting for the adoption to grow before investing even more engineering effort into this feature. Maybe worth opening a GitHub issue for that :)
          • wat10000 1 hour ago
            Right, nothing wrong with delegating the decision to a bunch of people who have thought long and hard about the best compromise, as long as it’s understood that it’s not perfect.
    • Arnt 2 hours ago
      Ah, let's have a long discussion of this.

      Unicode avoids "different" and "same", https://www.unicode.org/reports/tr15/ uses phrases like compatibility equivalence.

      The whole thing is complicated, because it actually is complicated in the real world. You can spell the name of Gießen "Giessen" and most Germans consider it correct even if not ideal, but spelling Massachusetts "Maßachusetts" is plainly wrong in German text. The relationship between ß and ss isn't symmetric. Unicode captures that complexity, when you get into the fine details.

    • mxmlnkn 2 hours ago
      I never understood why the recommended replacement for ß is ss. It is a ligature of sz (similar to & being a ligature of et) and is even pronounced ess-zet. The only logical replacement would have been sz, and it would have avoided the clash of Masse (mass) and Maße (measurements). Then again, it only affects whether the vowel before it is pronounced short or long, and there are better ways to encode that in written language in the first place.
    • looperhacks 2 hours ago
      MASS is allowed casing of Maß, but not the preferred casing: https://www.rechtschreibrat.com/DOX/RfdR_Amtliches-Regelwerk... Page 48
    • b2ccb2 3 hours ago
      The confusion likely stems from the relatively new introduction of the capitalized ẞ https://de.wikipedia.org/wiki/Gro%C3%9Fes_%C3%9F

      Maß capitalized (used to be) MASS.

      Funnily enough, Mass means one liter beer (think Oktoberfest).

      • andersa 2 hours ago
        It's strange, because I would expect "maß" as the case insensitive search to match "MASS" in the search text, but "mass" should not match "Maß".
      • looperhacks 2 hours ago
        Both Maß and Mass are valid spellings for a liter of beer ;) Not to confuse it with Maß, which just means any measurement, of course.